search

lck1201 / win_det_heatmaps

Window Detection in Facade Using Heatmaps Fushion
MIT License
48 stars 18 forks source link

Errors in loading staring dict: #8

Open Oushesh opened 2 years ago

Oushesh commented 2 years ago

RuntimeError: Error(s) in loading state_dict for PoseNet_2branch: Missing key(s) in state_dict: "backbone.conv1.weight", "backbone.bn1.weight", "backbone.bn1.bias", "backbone.bn1.running_mean", "backbone.bn1.running_var", "backbone.layer1.0.conv1.weight", "backbone.layer1.0.bn1.weight", "backbone.layer1.0.bn1.bias", "backbone.layer1.0.bn1.running_mean",

my command line was: python infer.py --cfg experiments/resnet/lr1e-3_x120-90-110_center_b2.yaml --model models/resnet18_model_latest.pth.tar --infer images So I am loading resnet18_model the configs from the .yaml file, and the model. Seems the definition or dictaionary does not contain those strings.

lck1201 commented 2 years ago

it looks strange only part of backbone modules can't be loaded.

Oushesh commented 2 years ago

yep. you got the same problem?