OpenNMT с Pytorch: cPickle.UnpicklingError: неверный ключ загрузки, ''

Я пытаюсь запустить проект OpenNMT, используя инструкцию по ссылке: http://forum.opennmt.net/t/text-summarization-on-gigaword-and-rouge-scoring/85/6

Я использую Python 2.7 и установил pytorch из репозитория github.
Я пытаюсь запустить программу, используя предварительно собранную модель 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.

1 ответ

Модель, которую вы скачали, предназначена для Lua-версии OpenNMT.

Если вы просто пользователь проекта, я рекомендую вам использовать эту версию, так как она наиболее поддерживаемая и стабильная.

Другие вопросы по тегам