Exporting weights of models #1003
Answered
by
FabianIsensee
paul-reiners
asked this question in
Q&A
-
|
Is it possible to export the weights of an nnU-Net model so that others can import them? Something like Torch Hub? |
Beta Was this translation helpful? Give feedback.
Answered by
FabianIsensee
Apr 8, 2022
Replies: 1 comment
-
|
Weights without the architecture to put them in are rather useless (unless you export as onnx). But sure you can do that. Just read the checkpoint files with torch.load. There is a dictionary in there where you will be able to find the model weights as well |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
FabianIsensee
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Weights without the architecture to put them in are rather useless (unless you export as onnx). But sure you can do that. Just read the checkpoint files with torch.load. There is a dictionary in there where you will be able to find the model weights as well