训练 SeGAN 模型时,以只读方式无法打开 <resnet-18.t7>

我正在尝试使用这里的存储库实现 SeGAN 论文。我安装了前提条件,并尝试使用以下命令训练模型:

th main.lua -baseLR 1e-3 -end2end -istrain "train"

然后我得到了这个错误:

警告:指定的初始化模型被忽略,因为将加载 resnet 模型!
/home/darya/distro/install/bin/luajit: cannot open <resnet-18.t7> in mode r  at /home/darya/distro/pkg/torch/lib/TH/THDiskFile.c:673
stack traceback:
    [C]: at 0x7f6f3df5e210
    [C]: in function 'DiskFile'
    /home/darya/distro/install/share/lua/5.1/torch/File.lua:405: in function 'load'
    /home/darya/SeGAN/networks/ModelTrainTestBatch.lua:285: in function 'LoadModel'
    main.lua:293: in main chunk
    [C]: in function 'dofile'
    ...rya/distro/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk

我不确定(因为我没有经验),但我认为 resnet 在阻止加载其他模型,但由于某些问题无法加载。我已经搜索过了,对我的问题没什么帮助。

非常感谢任何帮助。

点赞