Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Early stopping for pytorch training loop #65

Open
aiqc opened this issue Apr 15, 2021 · 0 comments
Open

Early stopping for pytorch training loop #65

aiqc opened this issue Apr 15, 2021 · 0 comments
Assignees
@aiqc
Labels
good first issue Good for newcomers

Comments

@aiqc
Copy link
Owner

aiqc commented Apr 15, 2021 •
edited
Loading

Difficulty: ★★☆☆☆

Background

We have the TrainingCallback class which has the Keras.MetricCutoff subclass for early stopping keras models. This is used intests.py module in the keras_image_binary class. there are many pytorch examples in tests.py

Problem

Models will either overtrain or just run for too long.

Solution

Since we run epochs manually in fn_train it should be easy enough to check the values and stop.
@aiqc aiqc added the good first issue Good for newcomers label Apr 15, 2021
@aiqc aiqc self-assigned this Apr 15, 2021
@aiqc aiqc changed the title Early stopping for pytorch Early stopping for pytorch training loop Apr 15, 2021
@aiqc aiqc removed their assignment Apr 15, 2021
@aiqc aiqc self-assigned this May 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@aiqc aiqc

Labels
good first issue Good for newcomers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@aiqc