修改args.py parser.add_argument('--resume', default=None, type=str, metavar='PATH', help='path to latest checkpoint')
default=None
model.load_state_dict(checkpoint[‘state_dict‘]) KeyError: ‘state_dict‘
原创
©著作权归作者所有:来自51CTO博客作者AI韬哥的原创作品,请联系作者获取转载授权,否则将追究法律责任
下一篇:cuda对应pytorh安装
提问和评论都可以,用心的回复会被更多人看到
评论
发布评论
相关文章
-
Error(s) pytorch 加载checkpoint state_dict出错:Missing key(s) && Unexpected key(s) in state_dict
ERROR:Traceback (most recent call last):File "test_0.py", line 130, in model = load_model()File "test_0.py", line 104, in load_modelmodel.load_state_dict(checkpoint['state_dict'])File "/home/...
pytorch 加载checkpoint 出错 Missing key(s)