corecai163 / odgnet Goto Github PK

2024-06-17 22:55:28,236 - UpTrans - INFO - Loading weights from ckpts/UpTrans/KITTI_models/ckpt-best.pth...
Traceback (most recent call last):
File "main.py", line 69, in <module>
main()
File "main.py", line 63, in main
test_net(args, config)
File "/home/brj/ODGNet/tools/runner.py", line 347, in test_net
builder.load_model(base_model, args.ckpts, logger=logger)
File "/home/brj/ODGNet/tools/builder.py", line 129, in load_model
base_model.load_state_dict(base_ckpt)
File "/home/brj/anaconda3/envs/IGEV_Stereo/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1604, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for UpTrans:
Missing key(s) in state_dict: "feat_extractor.refine_1.dict_embed.weight", "feat_extractor.refine_1.query.weight", "feat_extractor.refine_1.query.bias", "feat_extractor.refine_1.key.weight", "feat_extractor.refine_1.key.bias", "feat_extractor.refine_2.dict_embed.weight", "feat_extractor.refine_2.query.weight", "feat_extractor.refine_2.query.bias", "feat_extractor.refine_2.key.weight", "feat_extractor.refine_2.key.bias".
Unexpected key(s) in state_dict: "feat_extractor.refine_1.0.dict_embed.weight", "feat_extractor.refine_1.0.query.weight", "feat_extractor.refine_1.0.query.bias", "feat_extractor.refine_1.0.key.weight", "feat_extractor.refine_1.0.key.bias", "feat_extractor.refine_1.1.dict_embed.weight", "feat_extractor.refine_1.1.query.weight", "feat_extractor.refine_1.1.query.bias", "feat_extractor.refine_1.1.key.weight", "feat_extractor.refine_1.1.key.bias", "feat_extractor.refine_2.0.dict_embed.weight", "feat_extractor.refine_2.0.query.weight", "feat_extractor.refine_2.0.query.bias", "feat_extractor.refine_2.0.key.weight", "feat_extractor.refine_2.0.key.bias", "feat_extractor.refine_2.1.dict_embed.weight", "feat_extractor.refine_2.1.query.weight", "feat_extractor.refine_2.1.query.bias", "feat_extractor.refine_2.1.key.weight", "feat_extractor.refine_2.1.key.bias".