另外,模型训练完以后,在work_dir这个参数定义的目录下有个 output,里面会预存一份用于推理的模型checkpoint,可以直接使用。 参考这个链接:
https://github.com/modelscope/modelscope/blob/c32c0e5cfe053a6c5e18fdc2cbed4189fcec45e4/tests/trainers/test_trainer_with_nlp.py#L72
您好,如果您使用的是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