如何将Opus-MT模型转换为TensorFlow SavedModel格式?
将Opus-MT模型转换为TensorFlow SavedModel的可行性分析
Opus-MT项目提供了海量预训练神经翻译模型,它基于Marian-NMT构建——后者是纯C++实现的代码库,和nematus深度关联,而nematus本身就是基于TensorFlow开发的。
如果你出于部署需求想要把Opus-MT模型转换成TensorFlow SavedModel,这个方案完全具备可行性,核心依据如下:
- 预训练Opus-MT模型默认将各层权重存储为格式规范的
.npy文件,这种格式可直接被TensorFlow读取处理,为权重迁移打下了基础 - 用于训练新模型的Marian-NMT命令,能够生成与nematus兼容的模型格式,而nematus基于TensorFlow的特性,为模型向SavedModel转换打通了技术链路
内容的提问来源于stack exchange,提问作者lhk
相关产品推荐
相关产品推荐

