Skip to main content

模型报告

model_reporting.py 示例 展示了如何使用 OutputModel 类记录模型。

该示例执行以下操作:

  • examples项目中创建一个名为Model reporting example的任务。
  • 使用OutputModel对象注册先前训练的模型并记录其标签枚举。

初始化

为任务实例化了一个OutputModel对象。

from clearml import Task, OutputModel

# Connecting ClearML with the current process,
task = Task.init(project_name="examples", task_name="Model logging example")

# Create output model and connect it to the task
output_model = OutputModel(task=task)

标签枚举

使用OutputModel.update_labels()设置模型的标签枚举。

labels = {"background": 0, "cat": 1, "dog": 2}
output_model.update_labels(labels)

注册模型

使用OutputModel.update_weights()注册一个先前训练好的模型。 示例代码使用了存储在S3中的模型。

# Manually log a model file, which will have the labels connected above
output_model.update_weights(register_uri=model_url)

WebApp

模型出现在任务的ARTIFACTS标签中。

任务工件

点击模型名称将带您进入模型的页面,在那里您可以查看模型的详细信息并访问模型。

模型的LABELS标签显示其标签枚举。

模型标签选项卡

附加示例

请参阅PyTorch 模型更新以获取一个更健壮的示例,该示例训练一个模型,然后使用OutputModel记录它。