再训练以后,导出模型如何进行推理

咨询下,这个模型再训练以后产出 每次epoch_x.pth和output以后的bin模型,怎么进行推理,或者部署位推理服务

展开
收起
luanpeng 2023-03-02 17:37:53 663 分享 版权
1 条回答
写回答
取消 提交回答
  • 您好,推理的话直接把pipeline里面model_dir指定到output目录即可

    model_dir = '/xxxxxxx/output/' pipe = pipeline(Tasks.text_generation, model_dir=model_dir) print(pipe(input))

    2023-03-07 22:04:32
    赞同 展开评论

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

热门讨论

热门文章

还有其他疑问?
咨询AI助理