ML Katas

Visualize Training with TensorBoard

medium (<1 hr) pytorch training tensorboard visualization
this month by E

Integrate TensorBoard into a training loop:

  • Log training loss and validation accuracy.
  • Add histograms of weights and gradients.
  • Write a few sample images.

Open TensorBoard and verify logs.