V2B
V2B copied to clipboard
The provided pedestrian model CANNOT be successfully loaded
I run with following script:
python test_tracking.py --which_dataset KITTI --data_dir ./data/training/ --model_dir ./kitti_model/Pedestrian --model model.pth --category_name Pedestrian
And got following error:
RuntimeError: Error(s) in loading state_dict for Pointnet_Tracking: size mismatch for cml.conv4.conv3d.weight: copying a param with shape torch.Size([128, 64, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 64, 3, 1, 1]).