Implement Early Stopping
Add early stopping to a training loop:
- Monitor validation loss.
- Stop training if no improvement after 5 epochs.
- Save best model checkpoint.
Demonstrate on MNIST subset.
Add early stopping to a training loop:
Demonstrate on MNIST subset.