SwinUNETR/Pretrain: Fix resume mechanism #120

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

Open

razorx89 wants to merge 1 commit into Project-MONAI:main from razorx89:swinunetr-pretrain-fix-resume

razorx89 commented Sep 22, 2022 •

edited

Loading

Fixes #119.

Adds all relevant states to the checkpoint file.
Saves an additional "last.pt" file after each epoch.
Fixes state loading.


          Fix checkpoint content and loading from it

626c038

mingxin-zheng requested a review from dongyang0122

September 30, 2022 08:27

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

Labels

None yet