Problem when loading the weights

LiamLian0727 / USIS10K

[ICML 2024] Official repository of the paper: "Diving into Underwater: Segment Anything Model Guided Underwater Salient Instance Segmentation and A Large-scale Dataset"

Apache License 2.0

85 stars 10 forks source link

Problem when loading the weights #13

Closed likaiw2 closed 1 day ago

likaiw2 commented 4 days ago

When I am loading multi_class_model.pth for train, I met an error _pickle.UnpicklingError: invalid load key, '<'.

What may cause this problem?

LiamLian0727 commented 4 days ago

This may be because the model weights have not been downloaded. I haven't had this problem but after checking online I found similar cases, hope this helps!

likaiw2 commented 4 days ago

Hi, since I am abroad, it is hard for me to re-download from Baidu Disc, could you please provide Google Disc link for download?