pytorch预测的时候报异常了:
{RuntimeError}Error(s) in loading state_dict for DataParallel:
Unexpected key(s) in state_dict: "module.backbone.bn1.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.DownsampleUnit.bn1r.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.DownsampleUnit.bn2r.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.DownsampleUnit.bn3r.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.DownsampleUnit.bn1l.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.DownsampleUnit.bn2l.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.ShuffleUnit0_0.bn1.num_batches_tracked", "module.backbone.shuffles.ShuffleConvs_0.ShuffleUnit0_0.bn2.
Unexpected key(s) in state_dict: "module.backbone.bn1.num_batches_tracked"
最新推荐文章于 2024-01-09 12:16:17 发布
在使用PyTorch进行预测时遇到RuntimeError,错误信息提示state_dict中存在未预期的键,主要涉及到'num_batches_tracked'。问题出现在用不同版本的PyTorch(训练时为0.4.1,预测时为0.4.0)导致的加载模型异常。解决方法是通过修改加载字典的键,移除'module.'前缀。
摘要由CSDN通过智能技术生成