I met this error when run "python3 imagenet.py".

[bemoregt]

trafficstars

Hi, @Randl

I met this error when run "python3 imagenet.py --dataroot "pl_5plus3/" --resume "model_best.pth.tar" -e --scaling 0.5".

... ) number of parameters: 1372120 FLOPs: 40376352.0 => loading checkpoint 'model_best.pth.tar' Traceback (most recent call last): File "imagenet.py", line 215, in main() File "imagenet.py", line 152, in main model.load_state_dict(checkpoint['state_dict']) File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 769, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for ShuffleNetV2: Missing key(s) in state_dict: "conv1.weight", "bn1.weight", "bn1.bias", "bn1.running_mean", "bn1.running_var", "shuffles.ShuffleConvs_0.DownsampleUnit.conv1r.weight", "shuffles.ShuffleConvs_0.DownsampleUnit.bn1r.weight", "shuffles.ShuffleConvs_0.DownsampleUnit.bn1r.bias", "shuffles.ShuffleConvs_0.DownsampleUnit.bn1r.running_mean", "shuffles.ShuffleConvs_0.DownsampleUnit.bn1r.running_var", "shuffles.ShuffleConvs_0.DownsampleUnit.conv2r.weight", "shuffles.ShuffleConvs_0.DownsampleUnit.bn2r.weight", "shuffles.ShuffleConvs_0.DownsampleUnit.bn2r.bias", "shuffles.ShuffleConvs_0.DownsampleUnit.bn2r.running_mean", "shuffles.ShuffleConvs_0.DownsampleUnit.bn2r.running_var", "shuffles.ShuffleConvs_0.DownsampleUnit.conv3r.weight", "shuffles.ShuffleConvs_0.DownsampleUnit.bn3r.weight", "shuffles.ShuffleConvs_0.DownsampleUnit.bn3r.bias", "shuffles.ShuffleConvs_0.DownsampleUnit.bn3r.running_mean", "shuffles.ShuffleConvs_0.DownsampleUnit.bn3r.running_var", ... What;s wrong to me ??

Thanks in advance ~ Best, @bemoregt.