模型仓库简介

模型仓库是整个BML中模型的中央存储仓库，模型仓库可导入所有训练任务生成的模型并进行统一管理。

模型仓库支持的模型类型

从训练任务导入模型

BML支持多种训练方式，在训练任务执行成功后，可以将其生成的模型导入到模型仓库中进行模型的统一管理。

• 从预置模型调参导入模型

  • 在BML中通过预置模型调参完成模型训练后，可以将满足要求的模型发布到模型仓库中。
  • 一个预置模型调参项目的不同任务生成的模型只能发布到模型仓库中一个模型的不同版本下。

• 从Notebook导入模型

  • 在Notebook中保存模型并通过校验后，可以将其发布到模型仓库中。
  • 一个Notebook中保存的不同模型只能发布到模型仓库中一个模型的不同版本下。

• 从自定义作业导入模型

  • 自定义作业运行成功后，可以将其生成的模型发布到模型仓库中。
  • 一个自定义作业产生的模型只能发布到模型仓库中一个新的模型下。

从本地导入模型

BML也支持用户将本地已训练好的模型导入到模型仓库中，从而将模型部署为在线服务或转换为可行性离线部署的SDK等。

从本地导入时，BML当前支持的模型类型及相关框架如下所示：