tensorboard_data_server_quickstart_with_pytorch_summarywriter.py

1# 1. Install the data server alongside TensorBoard
2# pip install tensorboard tensorboard-data-server
3
4# 2. Use the standard TensorBoard Python API or CLI. 
5# TensorBoard will automatically detect and use the 
6# tensorboard-data-server binary if it is in your PATH.
7
8import os
9from torch.utils.tensorboard import SummaryWriter # or tensorflow.summary
10
11# Create a writer and log some dummy data
12writer = SummaryWriter("runs/quickstart")
13for i in range(100):
14    writer.add_scalar("Loss", 0.5 + (0.1 * i), i)
15writer.close()
16
17# 3. Launch TensorBoard from your terminal. 
18# The data server is used under the hood to index the 'runs' directory.
19# os.system("tensorboard --logdir=runs")