PyTorch TensorBoard Toy

The tensorboard_toy_pytorch.py example demonstrates the integration of Trains into code which creates a TensorBoard SummaryWriter object to log debug sample images. When the script runs, it creates an experiment named pytorch tensorboard toy example which is associated with the examples project.

Debug samples

The debug sample images appear in the RESULTS tab, DEBUG SAMPLES sub-tab.

  • test

  • test_gray

  • test_rgba

Hyperparameters

Command line arguments, which are automatically logged when argparse is used, and TensorFlow DEFINES, which are automatically logged when tensorflow is used, appear in the HYPER PARAMETERS tab.

Log

All other console output appears in the RESULTS tab, LOG tab.