Solved by imaluengo in post #9 
              
              
                Can you run, before you enter the training loop: 
net = net.float() 
It will transform the model parameters to float. 
And then in your training loop: 
z = net(x.float()) 
That should proceed without error. 
PS: replace .float() by .double() if you wish to have network + data in double precision for…