Should be implemented like batchnorm after the conv layer or before the input ? and what is the parameters to add in this case instead of m = nn.LayerNorm(input.size()[1:])
1 Like
Should be implemented like batchnorm after the conv layer or before the input ? and what is the parameters to add in this case instead of m = nn.LayerNorm(input.size()[1:])