DeyvidKochanov-TomTom / kprnet

MIT License
65 stars 12 forks source link

load model failed #7

Closed sunnyln closed 3 years ago

sunnyln commented 3 years ago

When I run the inference script, I got an error below:

(env3.7) lina@lina:~/python_code/kprnet-master$ python run_inference.py --checkpoint-path ./resnext_cityscapes_2p.pth --output-path ./results/ --semantic-kitti-dir /home/lina/python_code/SalsaNext-master/dataset/sequences/ resnext_cityscapes_2p.pth Traceback (most recent call last): File "run_inference.py", line 77, in main() File "run_inference.py", line 40, in main model.load_state_dict(torch.load(args.checkpoint_path)) File "/home/lina/venv/env3.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 830, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for DeepLabKP: Missing key(s) in state_dict: "kpclassifier.kpconv.weights", "kpclassifier.kpconv.kernel_points", "kpclassifier.bn.weight", "kpclassifier.bn.bias", "kpclassifier.bn.running_mean", "kpclassifier.bn.running_var", "final.weight", "final.bias".

could you please kindly tell me why, thx!

DeyvidKochanov-TomTom commented 3 years ago

Hi, these are the pre-training weights from cityscapes not the fully trained model. I can upload the full model later this week.