mbrossar / ai-imu-dr

AI-IMU Dead-Reckoning
MIT License
817 stars 225 forks source link

The checkpoint is not same with current model? Fail to load state_dict #75

Closed tuanndhust closed 1 year ago

tuanndhust commented 1 year ago

I got this error when testing the model.

RuntimeError: Error(s) in loading state_dict for TORCHIEKF: size mismatch for mes_net.cov_net.4.weight: copying a param with shape torch.Size([64, 32, 5]) from checkpoint, the shape in current model is torch.Size([32, 32, 5]). size mismatch for mes_net.cov_net.4.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([32]).