clearml-docs/docs/integrations/tensorboardx.md

2.1 KiB
Raw Blame History

title
TensorboardX

:::tip If you are not already using ClearML, see Getting Started. :::

TensorboardX is a data visualization toolkit to log information through PyTorch and visualize it through TensorBoard. ClearML automatically captures all data logged to TensorboardX, including scalars, images, video, plots, and text. All you have to do is add two lines of code to your script:

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

This will create a ClearML Task that captures your script's information, including Git details, uncommitted code, python environment, your TensorboardX metrics, plots, images, and text.

View the TensorboardX outputs in the WebApp, in the experiment's page.

TensorboardX WebApp scalars

Manual Logging

To augment its automatic logging, ClearML also provides an explicit logging interface.

See more information about explicitly logging information to a ClearML Task:

Examples

Take a look at ClearMLs TensorboardX examples:

  • TensorboardX with PyTorch - Demonstrates ClearML logging TensorboardX scalars, debug samples, and text in code using PyTorch
  • MegEngine MNIST - Demonstrates ClearML logging TensorboardX scalars in code using MegEngine
  • TensorboardX Video - Demonstrates ClearML logging TensorBoardX video data.