超级会员免费看
在使用PyTorch进行预测时遇到RuntimeError，错误信息提示state_dict中存在未预期的键，主要涉及到'num_batches_tracked'。问题出现在用不同版本的PyTorch（训练时为0.4.1，预测时为0.4.0）导致的加载模型异常。解决方法是通过修改加载字典的键，移除'module.'前缀。
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.num_batches_tracked", "module.backbone.shuffles.ShuffleC
订阅专栏 解锁全文
扫一扫
评论 4
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
