我是一个新的深度学习者,我试图重复https://github.com/swamiviv/LSD-seg中的代码,这是用于图像语义分割的,但在Pytorch中发生了一个问题,我真的需要你的帮助。在
非常感谢。在
以下是环境:
Ubuntu 14.04.1
Python 3.6.4
Pytorch 0.4.0
代码是用来下载预先训练好的模型的,但是我的服务器无法访问googledrive,所以我从其他方法下载了模型,并将模型文件更新到服务器上,并修改了相关代码,但是没有成功。在
原代码:
^{pr2}$我用其他方法下载了模型文件放到特定位置后,修改了代码:
import os.path as osp
import fcn
import torch
import torchvision
def VGG16(pretrained=False):
model = torchvision.models.vgg16(pretrained=False)
if not pretrained:
return model
state_dict = torch.load('/home/sunyuze/LSD/data/vgg16_from_caffe.pth')
model.load_state_dict(state_dict)
return model
错误警报为:
Traceback (most recent call last):
File "train.py", line 157, in <module>
main()
File "train.py", line 76, in main
vgg16 = torchfcn.models.VGG16(pretrained=True)
File "/home/sunyuze/LSD/code/torchfcn/models/vgg.py", line 17, in VGG16
state_dict = torch.load('/home/sunyuze/LSD/data/vgg16_from_caffe.pth')
File "/home/sunyuze/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 303, in load
return _load(f, map_location, pickle_module)
File "/home/sunyuze/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 459, in _load
magic_number = pickle_module.load(f)
_pickle.UnpicklingError: unpickling stack underflow
目前没有回答
相关问题 更多 >
编程相关推荐