Solved by Nikronic in post #2 
              
              
                Hi, 
PyTorch uses lower case for dimension in layers. 

Use nn.Conv2d instead. 
https://pytorch.org/docs/master/generated/torch.nn.Conv2d.html 
Bests