无法访问装载Lua的模型的参数

问题描述

我正在尝试运行基于神经网络的代码,该代码使用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 (将#修改为@)

相关问答

错误1:Request method ‘DELETE‘ not supported 错误还原:...
错误1:启动docker镜像时报错:Error response from daemon:...
错误1:private field ‘xxx‘ is never assigned 按Alt...
报错如下,通过源不能下载,最后警告pip需升级版本 Requirem...