Replies: 2 comments 1 reply
-
hi @rishiwobb When you download the pre-trained weight, you can find the download path which is usually the import timm
model = timm.create_model('resnet50', checkpoint_path='/home/hankyul/.cache/torch/hub/checkpoints/resnet50_a1_0-14fe96d1.pth') Thank you. hankyul |
Beta Was this translation helpful? Give feedback.
0 replies
-
Hi,@hankyul2 1- Build Model:
2- Save fine-tuned Model:
3- Load Model:
|
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Describe the bug
I wish to save a pre-trained model after calling the create_model method and then later on load the model from the file locally. How to do it?
Beta Was this translation helpful? Give feedback.
All reactions