模型报告
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记录它。