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

Add PyTorch Checkpoint Export and Load #1865

Open

patrickcleeve2 wants to merge 1 commit into MIC-DKFZ:master from patrickcleeve2:load_from_checkpoint

patrickcleeve2 commented Dec 28, 2023 •

edited

Loading

Similar to #1826, it would be conveient to be able to share a single .pt checkpoint file.

This PR adds:

Exporter cli for exporting a trained nnunet model as a single pytorch checkpoint file.
Reader for loading trained nnunet model from a single pytorch checkpoint file.

There is already the zip exporter, but that has a lot of additional data that isn't required if you only want to use the model for inference.

FabianIsensee self-assigned this


          add export to checkpoint, load from checkpoint

b49bdb8

patrickcleeve2 force-pushed the load_from_checkpoint branch from 4d67790 to b49bdb8 Compare

December 28, 2023 05:38

patrickcleeve2 marked this pull request as ready for review

December 28, 2023 05:38

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

Labels

None yet