corr_based_prediction
corr_based_prediction copied to clipboard
Does the model need accumulate the gradients?
At line 220 in main.py, the zero_grad is outside the training stage, the gradients will be accumulated along the training data.