运行结果及报错内容
本人使用的是python3.8,报错行代码为
net.load_state_dict(torch.load(model_path))
这个报错是为什么?
本人使用的是python3.8,报错行代码为
net.load_state_dict(torch.load(model_path))
参考https://blog.csdn.net/qq_27149279/article/details/106037681,改成net.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))