ljwztc / CLIP-Driven-Universal-Model

[ICCV 2023] CLIP-Driven Universal Model; Rank first in MSD Competition.
Other
521 stars 58 forks source link

Some doubts about validation and test #71

Closed 13772846495 closed 3 months ago

13772846495 commented 3 months ago

Firstly thanks for the great jobs. However, I encountered some issues during validation and testing. I only trained using the BTCV dataset. After training, I faced problems when using the trained epoch_xx.pth for validation.

RuntimeError: Error(s) in loading state_dict for Universal_model: Unexpected key(s) in state_dict: "", "0.weight", "0.bias", "2.weight", "2.bias", "3.weight", "3.bias", "weight", "bias". image

When I used backbone=’swinunetr’ and trained './out/swinunetr/epoch_xx.pth’, I encountered an error.

KeyError: ‘Key label of transform Invertd was missing in the data and allow_missing_keys==False.’

when trying to run 'pred_pseudo.py’. How can I resolve this issue?