跳到主要内容
版本:0.6.4

Starwhale 模型

overview

Starwhale 模型是一种机器学习模型的标准包格式,可用于多种用途,例如模型微调、模型评估和在线服务。 Starwhale 模型包含模型文件、推理代码、配置文件等等。

创建一个 Starwhale 模型

创建 Starwhale 模型有两种方法:通过 swcli 或通过 SDK

使用 swcli 创建 Starwhale 模型

使用 swcli 创建 Starwhale 模型之前,您可以定义一个model.yaml,其中描述了关于Starwhale模型的一些必要信息,然后运行以下命令:

swcli model build . --model-yaml /path/to/model.yaml

有关该命令和 model.yaml 的更多信息,请参阅swcli参考。需要注意的是,model.yaml 是非必要的。

使用 Python SDK 创建 Starwhale 模型

from starwhale import model, predict

@predict
def predict_img(data):
...

model.build(name="mnist", modules=[predict_img])

管理 Starwhale 模型

使用 swcli 管理 Starwhale 模型

命令说明
swcli model list列出项目中所有Starwhale模型
swcli model info显示有关Starwhale模型的详细信息
swcli model copy将Starwhale模型复制到另一个位置
swcli model remove删除Starwhale模型
swcli model recover恢复之前删除的Starwhale模型

使用 Web 界面管理 Starwhale 模型

管理 Starwhale 模型的历史版本

Starwhale 模型是版本化的。关于版本的基本信息可以参考 Starwhale中的资源版本控制

使用 swcli 管理 Starwhale 模型的历史版本

命令说明
swcli model history列出Starwhale模型的所有版本
swcli model info显示某个Starwhale模型版本的详细信息
swcli model diff比较两个版本的Starwhale模型
swcli model copy复制某个Starwhale模型版本到新的版本
swcli model remove删除某个Starwhale模型版本
swcli model recover恢复以前删除的Starwhale模型版本

模型评估

使用swcli进行模型评估

命令说明
swcli model run指定某个Starwhale模型进行模型评估

存储格式

Starwhale模型是一个打包了原始目录的tar文件。