Infer fails Нет модуля с именем «сохраненная_модель» - PullRequest
Новая
       45

Infer fails Нет модуля с именем «сохраненная_модель»

0 голосов
/ 09 декабря 2018

Я тренирую модель нейронного машинного перевода с opennmt-tf.В процессе обучения записан диск модели, и я хотел бы использовать модель для некоторого перевода, просто чтобы посмотреть, что получится.

Я пытаюсь использовать модель с этой командой 

onmt-main infer --features_file dummyIn.txt 
                --model run/export/latest/1544360647/saved_model.pb 
                --config default.yml

приводя к этой ошибке: 

WARNING:tensorflow:You provided a model configuration but a checkpoint already exists. The model configuration must define the same model as the one used for the initial training. However, you can change non structural values like dropout.
Traceback (most recent call last):
  File "/usr/local/bin/onmt-main", line 11, in <module>
    sys.exit(main())
  File "/usr/local/lib/python3.5/dist-packages/opennmt/bin/main.py", line 143, in main
    serialize_model=is_chief)
  File "/usr/local/lib/python3.5/dist-packages/opennmt/config.py", line 101, in load_model
    model = load_model_from_file(model_file)
  File "/usr/local/lib/python3.5/dist-packages/opennmt/config.py", line 44, in load_model_from_file
    module = load_model_module(path)
  File "/usr/local/lib/python3.5/dist-packages/opennmt/config.py", line 27, in load_model_module
    module = import_module(module_name)
  File "/usr/lib/python3.5/importlib/__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "<frozen importlib._bootstrap>", line 986, in _gcd_import
  File "<frozen importlib._bootstrap>", line 969, in _find_and_load
  File "<frozen importlib._bootstrap>", line 956, in _find_and_load_unlocked
ImportError: No module named 'saved_model'

Я не уверен, как интерпретировать эту ошибку - модель явно находится в указанном месте.

...