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 on_eval_epoch_end as a valid hook to TorchSnapshotSaver #826

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

dorukhansergin
Copy link
Contributor

Summary:
When TorchSnapshotSaver is used with save_every_n_eval_epochs > 0 and best_checkpoint_config, this hook is invoked.

https://www.internalfb.com/code/fbsource/[a8a4a7fba9a8a93af7382fa12e669c066f41024f]/fbcode/torchtnt/framework/callbacks/base_checkpointer.py?lines=273

However, it fails due to not being considered a valid hook. This diff fixes that

Differential Revision: D57083777

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D57083777

)

Summary:

When `TorchSnapshotSaver` is used with `save_every_n_eval_epochs > 0` and `best_checkpoint_config`, this hook is invoked.

https://www.internalfb.com/code/fbsource/[a8a4a7fba9a8a93af7382fa12e669c066f41024f]/fbcode/torchtnt/framework/callbacks/base_checkpointer.py?lines=273

However, it fails due to not being considered a valid hook. This diff fixes that

Differential Revision: D57083777
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D57083777

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants