What's new

Added 🎉

• Added StopEarlyCallback, a TorchTrainCallback for early stopping.
• Added parameter remove_stale_checkpoints to TorchTrainStep.

Changed ⚠️

• Minor changes to TorchTrainCallback interface.
• Weights & Biases TorchTrainCallback now logs best validation metric score.

Commits

c126442 torch train improvements (#64)