Even though I have explicitly tried moving both the model and the input tensors to GPU I get the following error. And on checking images.device and labels.device , it shows CPU
https://colab.research.google.com/drive/1Sgok4mHvuFT_YF3yHXWFR6RAZXVuCFRD 4
Help would be much appreciated.