You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
In "visualize_attention_map.ipynb", the trained model is loaded in the following line:
model.load_from(np.load("attention_data/Vit-B_16-224.npz"))
I used your train.py to finetune the Vit-B_16-224.npz with my custom data, which produced a Pytorch model checkpoint my-model.bin.
How do I perform the model.load_from for checkpoint my-model.bin? Do I need to convert the my-model.bin to a .npz format model? How can I convert it?
Thanks in advance!
The text was updated successfully, but these errors were encountered:
shenw000
changed the title
How to convert Pytorch model in .bin -> .npz ?
How to convert Pytorch model checkpoint in .bin -> .npz ?
Feb 18, 2023
Hello,
In "visualize_attention_map.ipynb", the trained model is loaded in the following line:
model.load_from(np.load("attention_data/Vit-B_16-224.npz"))
I used your train.py to finetune the Vit-B_16-224.npz with my custom data, which produced a Pytorch model checkpoint my-model.bin.
How do I perform the model.load_from for checkpoint my-model.bin? Do I need to convert the my-model.bin to a .npz format model? How can I convert it?
Thanks in advance!
The text was updated successfully, but these errors were encountered: