Skip to main content

PyTorch Lightning

pytorch-lightning 脚本展示了如何将ClearML集成到使用PyTorch Lightning的代码中。

示例脚本执行以下操作:

  • 在PyTorch内置的MNIST数据集上训练一个简单的深度神经网络
  • 定义Argparse命令行选项,这些选项会被ClearML自动捕获
  • examples项目中创建一个名为pytorch lightning mnist example的实验。

标量

测试损失和验证损失图会出现在ClearML网页界面的实验页面中的SCALARS部分。 资源利用率图,标题为:monitor: machine,也会出现在SCALARS标签中。所有这些 图表都是由ClearML自动捕获的。

PyTorch Lightning 标量

超参数

ClearML 自动记录使用 argparse 和 TensorFlow 定义定义的命令行选项,这些选项分别出现在 配置 > 超参数 > 参数TF_DEFINE 中。

PyTorch Lightning 参数

工件

实验创建的模型会出现在实验的ARTIFACTS标签中。

PyTorch Lightning 模型

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

控制台

所有其他控制台输出显示在控制台中。

PyTorch Lightning 控制台