开发者社区 > ModelScope模型即服务 > 自然语言处理 > 正文

模型训练完之后应该如何导出

用trainer.train()训练完之后,模型应该如何导出并使用?

展开
收起
1451997693938447 2022-11-21 23:28:16 1409 0
2 条回答
写回答
取消 提交回答
  • 另外,模型训练完以后,在work_dir这个参数定义的目录下有个 output,里面会预存一份用于推理的模型checkpoint,可以直接使用。 参考这个链接:

    https://github.com/modelscope/modelscope/blob/c32c0e5cfe053a6c5e18fdc2cbed4189fcec45e4/tests/trainers/test_trainer_with_nlp.py#L72

    2022-11-22 14:37:17
    赞同 展开评论 打赏
  • 您好,如果您使用的是PyTorch模型,您可以参考: https://www.modelscope.cn/docs/%E6%A8%A1%E5%9E%8B%E7%9A%84%E5%AF%BC%E5%87%BA 文档中描述的导出过程进行导出;如果是其他框架模型,ModelScope目前还没有统一的导出方案,您可以尝试使用https://github.com/onnx/tensorflow-onnx导出为ONNX,或使用tensorflow内置能力导出为graphdef

    2022-11-22 13:11:50
    赞同 展开评论 打赏

包含命名实体识别、文本分类、分词、关系抽取、问答、推理、文本摘要、情感分析、机器翻译等多个领域

热门讨论

热门文章

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载