Cannot load the checkpoint you provided.

[rayleizhu]

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]

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

[eraserNut]

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

[VladimirNikiforov]

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

[guanhuankang]

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!