我有一个为模型进行训练的管道,最后我想注册该模型,但我找不到任何相关信息。
以下是我的管道现在的样子:
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))
请按照文档进行跑步。