-
-
Notifications
You must be signed in to change notification settings - Fork 4.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[FEATURE] Early stopping when training #1991
Comments
Hi @saad-palapa , you could easily modify the train.py script adding your own EarlyStopping method, it's straightforward. You can use this simple yet effective implementation of EarlyStopping: There you'll find a simple example on how to use it (in the "Train the Model using Early Stopping" section): The provided training script in TIMM is just a general script, you have to adapt it to your case :D |
Is your feature request related to a problem? Please describe.
When running the training script, there is no way to stop training early when performance plateaus.
This causes 2 problems:
Describe the solution you'd like
I would like to add an early-stopping feature to the training script.
This would involve a few parameters:
Describe alternatives you've considered
Many other CV training platforms offer early stopping. I would like to have early stopping in TIMM because I prefer this repo over others.
The text was updated successfully, but these errors were encountered: