我可以使用 BERT 或 BART 进行机器翻译吗?

问题描述

我正在开展一个项目,以使用预先训练的模型并对其进行微调以进行自定义语言翻译,例如从英语到法语。是否可以在 Tensorflow 中加载这些模型并运行它们以查看翻译结果并随后进行微调?

解决方法

可能最快的方法是依赖 HuggingFace 转换器库。如果你不熟悉它,你可以看看他们的官方documentation。要针对 NMT 微调 ​​BART,您可以直接使用此提供的 script(它也适用于其他一些预训练模型)。