search

edwardzhou130 / Panoptic-PolarNet

Implementation for Panoptic-PolarNet (CVPR 2021)
BSD 3-Clause "New" or "Revised" License
166 stars 30 forks source link

RuntimeError: Error(s) in loading state_dict for ptBEVnet: #1

Closed coallar closed 3 years ago

coallar commented 3 years ago

Hello, I am trying to run the code using command: python3 test_pretrain.py,

but return: RuntimeError: Error(s) in loading state_dict for ptBEVnet: size mismatch for BEV_model.network.inc.conv.0.weight: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for BEV_model.network.inc.conv.0.bias: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for BEV_model.network.inc.conv.0.running_mean: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for BEV_model.network.inc.conv.0.running_var: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for BEV_model.network.inc.conv.1.conv1.0.weight: copying a param with shape torch.Size([64, 32, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 64, 3, 3]).

in fact, I did not change the code of the paper.