It would be nice to automatically stop once the test loss stops decreasing. Later we may want to add a simple learning rate scheduler but we don't need that right now.