search

nupurkmr9 / S2M2_fewshot

Other
112 stars 20 forks source link

link for pretrained model #14

Closed Yelysavite closed 3 years ago

Yelysavite commented 3 years ago

Thank you for your work. I've tried all the previous links。 I got a _pickle. unpicklingError when loading the pre-trained model while running train.py. I would like to ask if you can post a link to pretrained model. Thank you in advance.

nupurkmr9 commented 3 years ago

Hi, is this because of python version? I am able to load the models using python3.8 and it should work on python3.7 as well. Can you try on Python3.8?

ucasligang commented 3 years ago

No, it's your file had broken.

nupurkmr9 commented 3 years ago

Can you tell which model are you trying in torch.load()?

ucasligang commented 3 years ago

Sorry, I didn't check it carefully.