MindSpore ModelZoo Transformer 模型导出耗时久
收藏回复举报
MindSpore ModelZoo Transformer 模型导出耗时久
t('forum.solved') 已解决
发表于2024-03-30 09:08:07
0 查看

MindSpore ModelZoo,Transformer 模型:cid:link_0/tree/master/official/nlp/Transformer

迁移英中翻译模型至mindspore,当encoder及decoder层配置为24层

调整TransformerConfig.max_decode_length,随着TransformerConfig.max_decode_length值的增加(50-80-130),mindspore.export方法执行时间呈指数级增长

是否有优化措施缩短export时长

我要发帖子