Issue on evaluation - Githubissues

yijingru / Vertebra-Landmark-Detection

[ISBI 2020] Vertebra-Focused Landmark Detection for Scoliosis Assessment

MIT License

93 stars 26 forks source link

Issue on evaluation #6

Open wangcongbme opened 3 years ago

wangcongbme commented 3 years ago

Congratulations! Impressive work! However, I meet a problem when I try 'cobb_angle_eval.py'. The code tell me File "..../python3.7/site-packages/torch/nn/modules/module.py", line 1052, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for DenseNet: Missing key(s) in state_dict: "features.conv0.weight",......

Do you have any idea?It would be great if can reply when you have time!

yijingru commented 3 years ago

It looks like the layer names in the loaded weights do not fit your model. You may set load_state_dict(model, strict=False) to solve it.