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


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