How to convert Pytorch model checkpoint in .bin -> .npz ?

[shenw000]

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!