Thanks for the reply man, I didn’t get you sorry.
You mean, should I set device = ‘cpu’, which I already did actually. And the device output is ‘cpu’
01965×171 4.92 KB
I am a very newbie in Pytorch.
You can check my full code here, this will help you to understand better: https://gist.github.com/Dipeshpal/5d55e7016c5f4fcda4e5b5c459f8436d 15
Well, I am trying to load model using torch.load and model.load_state_dict both not working in Local machine with CPU (works on colab with GPU and CPU both)-
021557×713 69 KB
and
0021644×3296 255 KB
Sorry, if I write too much here. But this will help you to understand my problem. Thanks in advance dude.