V2B icon indicating copy to clipboard operation
V2B copied to clipboard

The provided pedestrian model CANNOT be successfully loaded

Open JunrQ opened this issue 3 years ago • 0 comments

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]).

JunrQ avatar Mar 06 '22 02:03 JunrQ