Azure ML CLI v2私有预览pipeline如何使用已注册模型作为输入
你可以直接使用如下yaml配置在pipeline中引用已注册的模型,运行时会自动将模型挂载到计算节点,访问逻辑和你当前使用的blob存储挂载完全一致:
1. 引用指定版本的已注册模型
inputs: input_model: data: azureml:<模型名称>:<模型版本号>
如果需要引用模型下的特定子路径,直接在版本号后拼接路径即可:
inputs: input_model: data: azureml:<模型名称>:<模型版本号>/weights/best.pt
2. 引用最新版本的已注册模型
将版本号替换为@latest即可自动拉取当前最新版本的模型:
inputs: input_model: data: azureml:<模型名称>@latest
你在组件代码中直接读取输入参数对应的挂载路径即可,无需做额外的模型拉取逻辑,和普通blob存储数据的访问方式完全相同。这个配置写法属于CLI v2 pipeline内置支持的schema,私有预览版本已经可以正常使用。
内容的提问来源于stack exchange,提问作者Francois
相关产品推荐
相关产品推荐

