C-HOI
C-HOI copied to clipboard
Missing Keys in State dict
When I try to run
python tools/test_pic.py configs/pic_v2.0/htc_rel_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn_20e_train_rel_dcn_semantichead.py pic_latest.pth --json_out det_result.json
I get the following output:
None
loading annotations into memory...
Done (t=0.00s)
creating index...
index created!
The model and loaded state dict do not match exactly
missing keys in source state_dict: reldn_head.freq_bias.rel_baseline.weight, reldn_head.prd_cls_feats.0.weight, reldn_head.prd_cls_feats.0.bias, reldn_head.prd_cls_feats.2.weight, reldn_head.prd_cls_feats.2.bias, reldn_head.prd_cls_scores.0.weight, reldn_head.prd_cls_scores.0.bias, reldn_head.spt_cls_feats.0.weight, reldn_head.spt_cls_feats.0.bias, reldn_head.spt_cls_feats.2.weight, reldn_head.spt_cls_feats.2.bias, reldn_head.spt_cls_scores.0.weight, reldn_head.spt_cls_scores.0.bias, reldn_head.prd_sbj_scores.0.weight, reldn_head.prd_sbj_scores.0.bias, reldn_head.prd_sbj_scores.2.weight, reldn_head.prd_sbj_scores.2.bias, reldn_head.prd_obj_scores.0.weight, reldn_head.prd_obj_scores.0.bias, reldn_head.prd_obj_scores.2.weight, reldn_head.prd_obj_scores.2.bias, reldn_binary_head.vis_rank_feats.0.weight, reldn_binary_head.vis_rank_feats.0.bias, reldn_binary_head.vis_rank_feats.2.weight, reldn_binary_head.vis_rank_feats.2.bias, reldn_binary_head.spt_rank_feats.0.weight, reldn_binary_head.spt_rank_feats.0.bias, reldn_binary_head.spt_rank_feats.2.weight, reldn_binary_head.spt_rank_feats.2.bias, reldn_binary_head.proj.weight, reldn_binary_head.proj.bias, reldn_binary_head.readout.0.weight, reldn_binary_head.readout.0.bias
System: Ubuntu 16.04 Cuda 9.2 Nvidia Driver 396.37
Please help!