Tensorboard (experimental) in PyTorch 1.1

pip install --upgrade torch
pip install tb-nightly
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir='./logs')# Track variables (e.g. loss, kld, etc.) that change
writer.add_scalar('KL Divergence', kl_div)
writer.add_scalar('Total Loss', loss)
writer.add_text('Decoder output', 'Some output')
tensorboard --logdir=logs
PyTorch 1.1 with Tensorboard ❤

--

--

--

Artificial Intelligence Researcher | Developer

Love podcasts or audiobooks? Learn on the go with our new app.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Akash Raj

Akash Raj

Artificial Intelligence Researcher | Developer

More from Medium

AI-Assisted Annotation with Multimodal Neural Networks

Multi GPU Training In Pytorch

Capture Pytorch concept for better understanding

Validating SMILES with RDKit, PySMILES, MolVS, and PartialSMILES