cbd:

Looking after may topic on forum i come to know that model should be run on GPU. But it is already runs on GPU as i print it in forward function.


Hey can you do one more thing print any layer weights in your model before passing the input.
Eg: print(net.conv1.weight) . It will print your weights but at the end you will also see device='cuda:0', requires_grad=True) which means device weights are in GPU cuda.