Tensorborard

Weight&Biases

torch.utils.tensorboard - PyTorch 1.10.1 documentation

Tensorboard

• scalar : metric등 상수 값의 연속(epoch) 표시
• graph : 모델의 computational graph 표시
• histogram : weight 등 값의 분포를 표현
• image ,mesh etc

#make directory for tensorboard
import os
logs_base_dir = "logs"
os.makedirs(logs_base_dir, exist_ok=True)

from torch.utils.tensorboard import SummaryWriter
import numpy as np

 # Writer will output to ./runs/ directory by default if dir not given
writer = SummaryWriter(log_base_dir)  
for n_iter in range(100):
    writer.add_scalar('Loss/train', np.random.random(), n_iter)
    writer.add_scalar('Loss/test', np.random.random(), n_iter)
    writer.add_scalar('Accuracy/train', np.random.random(), n_iter)
    writer.add_scalar('Accuracy/test', np.random.random(), n_iter)

Weight&Biases