文档仅提及使用命令行来运行translate.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
我找不到有关如何执行此操作的任何文档。请帮忙!