Cannot load the checkpoint you provided.

eraserNut / MTMT

Code for the CVPR 2020 paper "A Multi-task Mean Teacher for Semi-supervised Shadow Detection"

101 stars 14 forks source link

Cannot load the checkpoint you provided. #19

Open rayleizhu opened 3 years ago

rayleizhu commented 3 years ago

I use the trained model you provided in BaiduNetDisk (iter_10000.pth)

It seems your model doesn't match the checkpoint, so that the following error occurs:

RuntimeError: Error(s) in loading state_dict for TUN_bone:
        Unexpected key(s) in state_dict: "merge1.number_per_fc.weight", "merge1.number_per_fc.bias".

shilin21 commented 3 years ago

Same problem, your iter_10000.pth cannot load. Could you please check it?

eraserNut commented 3 years ago

I update some file. The old file is the ablation version and i modify it.

VladimirNikiforov commented 2 years ago

Hi @eraserNut, could you please upload checkpoint file to Google Drive as well?

guanhuankang commented 2 years ago

Hi, could you please upload the checkpoint file to Google Drive? It seems like the file in BaiduNetDisk can not be downloaded. Thank you in advances!