torch0.4.0版本加载torch0.4.1版本训练的pytorch模型
利用torch0.4.0版本的pytorch加载在torch0.4.1版本上训练的pytorch模型时,会遇到如下报错:
RuntimeError: Error(s) in loading state_dict for DataParallel:
Unexpected key(s) in state_dict: "xxxxx.num_batches_tracked"
如图所示:
此时只需...
原创
2019-06-14 16:27:56 ·
1942 阅读 ·
1 评论