模型仓库简介
更新时间:2020-11-26
模型仓库是整个BML中模型的中央存储仓库,模型仓库可导入所有训练任务生成的模型并进行统一管理。
模型仓库支持的模型类型
技术方向 | 类型 |
视觉 | 图像分类 |
物体检测 | |
实例分割 | |
自然语言处理 | 文本分类 |
短文本相似度 | |
序列标注 | |
文本实体抽取 | |
通用 | N/A |
从训练任务导入模型
-
从脚本调参导入模型
- 在BML中通过脚本调参完成模型训练后,可以将满足要求的模型发布到模型仓库中
- 一个脚本调参的不同任务的模型只能发布到模型仓库中的一个模型下
-
从Notebook导入模型
- 在Notebook中保存模型并通过校验后,可以将其发布到模型仓库中
- 一个Notebook中保存的不同模型只能发布到模型仓库中的一个模型下
-
从自定义作业导入模型
- 自定义作业运行成功后,可以将其生成的模型发布到模型仓库中
- 一个自定义作业产生的模型只能发布到模型仓库中一个新的模型下