torch에서 Tensorboard를 이용해서 loss plot 하는 방법

torch에서 Tensorboard를 이용해서 loss plot 하는 방법

사용자 입장에서 파이토치 라이브러리를 사용하면서 필요한 부분에 대하여 포스팅 한 글 입니다. 

Plot training loss with tensorboard

---

1. 파이썬 코드에서 SummaryWriter를 import 해줍니다. 

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

2. Training에서 Loss를 계산하고
    아래 코드를 이용해서 로컬로 파일을 써줍니다. 

아마 다른 코드가 궁금할 수 있는데, writer를 다른 코드와 dependency가 없이 값을 써주는 것 입니다. 

따라서 원하는 위치에 적어주시면 됩니다. 

writer.add_scalar("Loss/train", loss.item(), i)

3. 훈련코드를 돌려주면서 ananconda prompt 창을 켭니다. 

 

jupyter notebook이 있는 폴더에서 다음과 같이 커맨드를 입력해줍니다. 

tensorboard --logdir runs

--logdir <Directory 위치> 를 입력해주고 chrome에서 localhost:6006을 입력해서 접속해줍니다.  
runs 라고 입력한 이유는 torch에서 runs 이름의 디렉토리를 생성하기 때문입니다. 

---

python 3.6.8
pytorch 1.7.0
numpy 1.19.2