PyTorch 문서) PyTorch Recipes - How to use TensorBoard with PyTorch

파이토치와 함께 텐서보드를 사용하는 방법

• 원글 : How to use TensorBoard with PyTorch
• 텐서보드(Tensorboard)는 머신러닝 실험에 사용되는 시각화 툴킷이다.
  • 손실값, 정확도와 같은 평가 메트릭을 시각화(visualization)하거나 추적(tracing)할 수 있고
  • 모델 구조를 모델 그래프(model graph)로써 시각화할수 있으며
  • 그외에 히스토그램, 이미지 표시 등 다양한 기능을 제공한다.
• 이 글에서는 텐서보드 설치에서부터 파이토치와 함께 활용하는 기본 사용법, 로그로 기록한 데이터를 텐서보드 UI에서 시각화 하는 방식을 설명한다.

Installation

• 우선 파이토치가 설치되어서, 모델과 메트릭에 대한 로그를 텐서보드 로그 디렉토리(Tensorboard log directory)에 남길 수 있어야 한다.

Anaconda를 통한 PyTorch 설치(추천)

$ conda install pytorch torchvision -c pytorch

pip을 통한 PyTorch설치

$ pip install torch torchvision

Using TensorBoard in PyTorch

• 시각화할 대상에 대한 로그를 남기기 위해서는 우선, SummaryWriter 객체를 생성해야 한다.

import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()

• 위에서 생성한 writer는 출력을 남기는 디렉토리의 기본 설정이 ./runs/ 디렉토리로 되어 있다.

Log scalars

•