Python:RuntimeError报错是为什么?

运行结果及报错内容

本人使用的是python3.8,报错行代码为

net.load_state_dict(torch.load(model_path))

img


这个报错是为什么?

参考https://blog.csdn.net/qq_27149279/article/details/106037681,改成net.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^