Loss not showing - pytorch

[Anand_Deep]

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

Notebook Link
(Section : CNN Pytorch)

[Ishvinder]

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.

[Anand_Deep]

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

[Ishvinder]

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