corr_based_prediction icon indicating copy to clipboard operation
corr_based_prediction copied to clipboard

Does the model need accumulate the gradients?

Open aflyingnoob opened this issue 5 years ago • 0 comments

At line 220 in main.py, the zero_grad is outside the training stage, the gradients will be accumulated along the training data.

aflyingnoob avatar Aug 18 '20 06:08 aflyingnoob