Solved by skerlet_flandorle in post #3 
              
              
                m = nn.GroupNorm(1, 6) 
output = m(input) 
When do you use output = m(input)? 
In one sample, x=Relu(GroupNorm(CNN(x))) 
However, I do not know whether to do it for each CNN layer or only for the last layer.