问题描述
我正在尝试运行基于神经网络的代码,该代码使用Torchfile包中的加载功能加载Vgg16模型。代码使用此文件(https://github.com/zhanghang1989/PyTorch-Multi-Style-Transfer/blob/master/experiments/utils.py)中的此功能来加载vgg16模型:
def init_vgg16(model_folder):
"""load the vgg16 model feature"""
if not os.path.exists(os.path.join(model_folder,'vgg16.weight')):
if not os.path.exists(os.path.join(model_folder,'vgg16.t7')):
os.system(
'wget http://cs.stanford.edu/people/jcjohns/fast-neural-style/models/vgg16.t7 -O ' + os.path.join(model_folder,'vgg16.t7'))
vgglua = load_lua(os.path.join(model_folder,'vgg16.t7'))
vgg = Vgg16()
for (src,dst) in zip(vgglua.parameters()[0],vgg.parameters()):
dst.data[:] = src
torch.save(vgg.state_dict(),os.path.join(model_folder,'vgg16.weight'))
这会导致以下错误:
in init_vgg16
for (src,vgg.parameters()):
TypeError: 'NoneType' object is not callable
我真的不知道为什么会出现此错误。我在googleColab中使用的Torchfile软件包版本是0.1.0,而Torch是0.3.1。
请帮忙吗?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)