我正在尝试使用链接中的指令运行OpenNMT项目:http://forum.opennmt.net/t/text-summarization-on-gigaword-and-rouge-scoring/85/6

我正在使用Python 2.7并从github存储库安装了pytorch。
我试图使用OpenNMT的预构建模型运行该程序,该模型已从以下位置下载:http://opennmt.net/Models/

我尝试了命令:

python translate.py -model textsum_epoch7_14.69_release.t7 -src data/Giga/input.txt


出现以下错误:

Traceback (most recent call last):
  File "translate.py", line 151, in <module>
    main()
  File "translate.py", line 70, in main
    translator = onmt.Translator(opt)
  File "/home/ubuntu/opennmt/onmt/Translator.py", line 21, in __init__
    checkpoint = torch.load(opt.model)
  File "/usr/local/lib/python2.7/dist-packages/torch/serialization.py", line 229, in load
    return _load(f, map_location, pickle_module)
  File "/usr/local/lib/python2.7/dist-packages/torch/serialization.py", line 367, in _load
    magic_number = pickle_module.load(f)
cPickle.UnpicklingError: invalid load key, ''.


请让我知道我需要做什么,以便可以使用该模型并检查库OpenNMT。

最佳答案

您下载的模型适用于OpenNMT的Lua version

如果您只是该项目的用户,我建议您使用此版本,因为它是最受支持和最稳定的版本。

10-06 06:28