Loss not showing - pytorch

I am trying to plot the accuracy but it’s giving me tensor sizes :confused:

Notebook Link
(Section : CNN Pytorch)

You must have had a print statement in training loop or some function which is called.
Please check once, and the notebook is private, please give it a public access.

1 Like

https://colab.research.google.com/drive/1kPDvBGqBs7KEglv5gd3_PtYtCnm2lTnL?usp=sharing

Here’s the print statements in forward function in net class.
You’ll have to remove them first