EarlyStopping restore_best_weights argument similar to Keras
🚀 Feature
EarlyStopping argument restore_best_weights restores model weights from the epoch with the best value of the monitored loss, similar to Keras. Would not need to run ModelCheckpoint every epoch.
Motivation
Good to have to avoid even more pytorch boilerplate
Alternatives
Calling ModelCheckpoint ever epoch