我们已经为您准备好复现该模型所需的环境,您可以创建实例、一键复用哦!

模型的使用文档涵盖模型所需环境、模型简介以及复现模型的具体步骤。模型效果和模型详细来源可以点击论文地址和模型来源链接查看哦~

Opennmt

2022-02-15 model:opennmt-ubuntu18.04-cuda10.2-cudnn7-python3.7-pytorch1.6.0-new

模型标签
文本分类文本匹配文本生成文本摘要

模型简介

OpenNMT 是一个由 Harvard NLP (哈佛大学自然语言处理研究组) 开源的 Torch 神经网络机器翻译系统。简单的通用接口,只需要源/目标文件。快速高性能GPU训练和内存优化。提高翻译性能的最新的研究成果。可配对多种语言的预训练模型。允许其他序列生成任务的拓展,如汇总和图文生成。 机器翻译,又称为自动翻译,是利用计算机将一种自然语言(源语言)转换为另一种自然语言(目标语言)的过程。它是计算语言学的一个分支,是人工智能的终极目标之一。



1. 使用方法

1.1. 模型训练

在 root/OpenNMT-py(默认)路径下,输入

onmt_train -config toy_en_de.yaml
参数 说明
-config 配置文件

1.2. 模型测试

onmt_translate -model toy-ende/run/model_step_1000.pt -src toy-ende/src-test.txt -output toy-ende/pred_1000.txt -gpu 0 -verbose
参数 说明
-model 模型
-src 源文件
-gpu 0 第一张显卡

注:预测命令运行结束后,会在toy-ende/pred_1000.txt文件下输出运行结果。

2. 模型来源

https://github.com/OpenNMT/OpenNMT-py