[Pytorch] TensorboardX

Tensorboard는 원래 tensorflow에서만 지원하던 데이터 흐름을 확인하기 위한 소프트웨어 프로그램이다. Pytorch도 지원이 된지는 오래된거 같은데 이제라도 알았으니 적극적으로 사용해야겠다.

 

가장 큰 장점 중 하나는 실시간성이다. 보통 loss나 특징 metric지표들을 실험이 모두 종료된 후에 보기 마련인데, tensorboard를 사용하면 실시간으로 loss가 어떻게 그려지는지 확인할 수 있다. 수치값도 좋지만 눈으로 확인 할 수 있다는 부분이 이해하기도 편하다.

 

sin, cos 그래프 데이터를 tensorboardX로 살펴보자. 보통 log형식으로 파일에 입력하는 부분에 tensorboardX 코드도 아래와 같은 형식으로 작성하면된다. 다양한 기능은 Docs를 참고하자.

https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html

Visualizing Models, Data, and Training with TensorBoard — PyTorch Tutorials 2.0.0+cu117 documentation

import math
from tensorboardX import SummaryWriter

writer = SummaryWriter()

for step in range(-360, 360):
    angle_rad = step * math.pi / 180
    writer.add_scalar('sin',math.sin(angle_rad),step)
    writer.add_scalar('cos',math.cos(angle_rad),step)
    
writer.close()

 

터미널에서 먼저 tensorboard를 실행한다. tensorboard에서 자동적으로 데이터를 fetch해오기 때문에 tensorboard 실행동안 파일을 실행시키기만 하면 된다.

tensorboard --logdir=runs

 

이제 위의 sin, cos 파일을 실행하면 입력결과가 tensorboard로 가져와진다.

python example.py

적극 활용하도록 하자-!