Works like a charm. Thanks @awaelchli!
The only other instance of on_training_epoch_end I could find is in the paragraph above the code here in the docs
1 Like
Works like a charm. Thanks @awaelchli!
The only other instance of on_training_epoch_end I could find is in the paragraph above the code here in the docs