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.

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

add GetBestPtCheckpointJob #432

Merged
merged 3 commits into from
Jul 26, 2023
Merged

add GetBestPtCheckpointJob #432

merged 3 commits into from
Jul 26, 2023

Conversation

vieting
Copy link
Contributor

@vieting vieting commented Jul 25, 2023

We have the GetBestTFCheckpointJob and I'd like to add the equivalent for torch checkpoints. I added a very similar GetBestPtCheckpointJob since this does not interfere with existing setups. We might also think about how to avoid some of the code duplications, but this seems to be the simplest solution.

Copy link
Contributor

@JackTemaki JackTemaki left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think for PyTorch we can simplify this a little bit more

returnn/training.py Outdated Show resolved Hide resolved
returnn/training.py Outdated Show resolved Hide resolved
Copy link
Contributor

@JackTemaki JackTemaki left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think for PyTorch we can simplify this a little bit more

@vieting vieting requested a review from JackTemaki July 25, 2023 16:20
@vieting
Copy link
Contributor Author

vieting commented Jul 25, 2023

I think for PyTorch we can simplify this a little bit more

Right. In this case, the code duplication is also reduced, so we can just add the torch specific job imho.

@vieting vieting requested a review from michelwi July 26, 2023 14:24
@vieting vieting merged commit 1f7bc8a into main Jul 26, 2023
4 checks passed
@vieting vieting deleted the peter_best_pt_ckpt branch July 26, 2023 14:32
Atticus1806 pushed a commit that referenced this pull request Oct 2, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants