[Pytorch] 텐서보드 초간단 사용법

SummaryWriter 임포트

from torch.utils.tensorboard import SummaryWriter

 

Writer 생성

writer = SummaryWriter()

위와 같이 writer을 생성하면 기본값으로 ./runs에 로그가 저장된다. 저장경로를 바꾸고 싶으면 다음과 같이 하면된다. 

writer = SummaryWriter(log_dir)

 

로그 찍을 변수 넣어주기

writer.add_scalar('Loss/train', train_loss, epoch)

원하는 이름을 적고, 해당 이름에 대한 값을 전달해준 뒤에, 어떤 step에 대해서 로그를 찍을 것인지 epoch를 전달해주었다. 

 

Writer 닫기

writer.close()

 

텐서보드 실행

tensorboard --logdir=runs

logdir를 지정해주고 텐서보드를 실행합니다. 학습 전에 실행해도되고 후에 실행해도됩니다. 

 

 

텐서보드 접속

http://localhost:6006 

이제 여기 들어가면 log를 시각화해준것을 볼 수 있습니다.