Submitted by Rishh3112 t3_120gvgw in deeplearning
Rishh3112 OP t1_jdhiguj wrote
Reply to comment by trajo123 in Cuda out of memory error by Rishh3112
i just checked in my training loop I'm using loss.item()
_vb__ t1_jdiwjqk wrote
Are you calling the zero_grad method on your optimizer in every step of your training loop?
Viewing a single comment thread. View all comments