azure机器学习服务从管道注册模型(CLI v2)



我有一个为模型进行训练的管道,最后我想注册该模型,但我找不到任何相关信息。

以下是我的管道现在的样子:

type: pipeline_job
compute:
target: azureml:gpu-cluster
inputs:
input_images:
data:
datastore: azureml:dualcam
path: /image-20210701*
jobs:
training:
type: component_job
component: file:./components/training.yml
inputs:
input_images: inputs.input_images

您可以使用upload_file将其上传到运行中,然后注册模型。

在尝试注册模型之前,将模型上载到运行历史记录中:

run.upload_file("outputs/my_model.pickle", "outputs/my_model.pickle")
print(run.get_file_names())
print(os.getcwd())
model = run.register_model(model_name='simtemp', model_path="outputs/{}".format(model_file_name))

请按照文档进行跑步。

相关内容

  • 没有找到相关文章