Solved by ptrblck in post #2 
              
              
                Some layers such as batchnorms and the bias parameter of other layers cannot be used in init methods, which require a multi-dimensional parameter. 
The weight of nn.BatchNorm2d is by default now initialized with ones, while the bias with zeros, which seems to be the recommended way now. 
Pooling lay…