Skip to main content

TensorboardX

tip

如果您还没有使用ClearML,请参阅入门指南

TensorboardX 是一个数据可视化工具包,用于通过 PyTorch 记录信息并通过 TensorBoard 进行可视化。 ClearML 自动捕获所有记录到 TensorboardX 的数据,包括标量、图像、视频、图表和文本。你只需要在你的脚本中添加两行代码:

from clearml import Task

task = Task.init(task_name="<task_name>", project_name="<project_name>")

这将创建一个ClearML任务,该任务将捕获您的脚本信息,包括Git详细信息、未提交的代码、Python环境、您的TensorboardX指标、图表、图像和文本。

在实验页面中查看WebApp中的TensorboardX输出。

TensorboardX WebApp 标量

自动日志控制

默认情况下,当ClearML集成到您的脚本中时,它会捕获所有的TensorboardX图表、图像、指标、视频和文本。 但是,您可能希望对实验日志的内容有更多的控制。

要控制任务的框架日志记录,请使用Task.init()auto_connect_frameworks参数。 通过将参数设置为False,可以完全禁用所有自动日志记录。为了更精细地控制记录的框架,可以输入一个包含框架-布尔对的字典。

例如:

auto_connect_frameworks={
'tensorboard': False,'matplotlib': False, 'tensorflow': False, 'pytorch': True,
'xgboost': False, 'scikit': True, 'fastai': True, 'lightgbm': False,
'hydra': True, 'detect_repository': True, 'tfdefines': True, 'joblib': True,
'megengine': True, 'catboost': True
}

请注意,tensorboard 键用于启用/禁用 TensorboardXTensorBoard 的自动日志记录。

手动日志记录

为了增强其自动日志记录功能,ClearML 还提供了一个显式的日志记录接口。

查看有关显式记录信息到ClearML任务的更多信息:

示例

看看ClearML的TensorboardX示例:

  • TensorboardX with PyTorch - 演示了在代码中使用PyTorch记录TensorboardX标量、调试样本和文本的ClearML日志
  • MegEngine MNIST - 演示了在代码中使用MegEngine记录TensorboardX标量的ClearML日志
  • TensorboardX 视频 - 演示 ClearML 记录 TensorBoardX 视频数据。