文档只提到使用命令行来运行repotranslate.py
中已经给出的脚本OpenNMT-py
,即使用
python translate.py -model model_name.pt -src data/src-test.txt -output pred.txt
这很好,但是我如何使用该onmt
库将经过训练的模型加载到 jupyter 笔记本中并在旅途中进行翻译?我希望它能够在 jupyter 中做这样的事情:
model = onmt.module.load(/path/to/model.pt)
text = "hola como estas"
model.translate(text)
Output:
>>>Hello how are you
我找不到任何有关如何执行此操作的文档。请帮忙!