You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
outputs in training_epoch_end contain only outputs from last batch repeated multiple times. I believe it got broken only in 1.4.0, but in 1.3.x it worked.
🐛 Bug
outputs in training_epoch_end contain only outputs from last batch repeated multiple times. I believe it got broken only in 1.4.0, but in 1.3.x it worked.
To Reproduce
This will print the same loss repeated 10 times (equal to last batch loss) in training_epoch_end:
Expected behavior
Output from all steps/batches is available in training_epoch_end (not only from last batch)
Environment
The text was updated successfully, but these errors were encountered: