RuntimeError: Error(s) in loading state_dict for BNInception:

[javastudent12345]

RuntimeError: Error(s) in loading state_dict for BNInception: size mismatch for conv1_7x7_s2_bn.weight: copying a param with shape torch.Size([1, 64]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for conv1_7x7_s2_bn.bias: copying a param with shape torch.Size([1, 64]) from checkpoint, the shape in current model is torch.Size([64]).

How to solve this problem？

[aicentral]

I have the same error but for InceptionV3 the model Error(s) in loading state_dict for InceptionV3: size mismatch for conv_batchnorm.weight: copying a param with shape torch.Size([1, 32]) from checkpoint, the shap e in current model is torch.Size([32]).

command used python test_video.py --arch InceptionV3 --dataset moments --weights pretrain/TRN_moments_RGB_InceptionV3_TRNmultiscale_segment8_best.pth.tar --video_file sample_data/input.mp4 --rendered_output sample_data/output.mp4

env. details python 3.8 Cuda 10.1.243 cudnn 7.6.3 pytorch 1.4.0 py3.8_cuda10.1.243_cudnn7.6.3_0
torchvision 0.5.0 py38_cu101