Skip to content

SwinUNETR/Pretrain: Fix resume mechanism#120

Open
razorx89 wants to merge 1 commit into
Project-MONAI:mainfrom
razorx89:swinunetr-pretrain-fix-resume
Open

SwinUNETR/Pretrain: Fix resume mechanism#120
razorx89 wants to merge 1 commit into
Project-MONAI:mainfrom
razorx89:swinunetr-pretrain-fix-resume

Conversation

@razorx89

@razorx89 razorx89 commented Sep 22, 2022

Copy link
Copy Markdown

Fixes #119.

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

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.

SwinUNETR/Pretrain: Resume does not work

1 participant