文件名称:翻译:翻译-PyTorch语言库
文件大小:274KB
文件格式:ZIP
更新时间:2024-02-23 22:26:04
machine-learning pytorch artificial-intelligence onnx MachinelearningPython
注意 现在不建议使用PyTorch Translate,请改用 。 翻译-PyTorch语言库 翻译是一个用PyTorch编写的机器翻译库。 它提供了序列到序列模型的训练。 Translate依赖于 (一个通用的序列到序列库),这意味着可以训练在Translate和Fairseq中实现的模型。 Translate还提供了通过将某些模型导出到Caffe2图形以及从C ++加载和运行这些模型以用于生产目的的能力。 当前,我们将组件(编码器,解码器)分别导出到Caffe2,并且波束搜索是在C ++中实现的。 在不久的将来,我们也将能够导出波束搜索。 我们还计划为更多型号增加出口支持。 快速开始 如果您只想训练/评估MT模型,而不想通过ONNX将模型导出到Caffe2,则可以按照以下几步安装Translate for Python 3: 克隆此存储库git clone https://github.com/pytorch/translate.git pytorch-translate && cd pytorch-translate 运行python setup.py install 如果