服务器没联网,load_state_dict_from_url失败

问题

服务器由于网络问题,没法load_state_dict_from_url

解决办法

通过本地网络把url内容下载到本地,再上传到服务器。
代码替换
s = load_state_dict_from_url(url, progress=progress)
通过url获取字典,返回对象为字典。
把代码替换为
s = torch.load('/home/xiezhiheng/res20_cifar10/model/resnet20_cifar10.th')
其中load的文件是上面url对应的文件。
两行代码获取的s是等价的。

posted @ 2024-04-24 19:56  心比天高xzh  阅读(105)  评论(0)    收藏  举报