Solved by lxtGH in post #2 
              
              
                for m in self.modules():
			if isinstance(m, nn.Conv2d):
				nn.init.normal(m.weight, mean=0, std=0.01)
				nn.init.constant(m.bias, 0)