Machine Learning/Pytorch1 Pytorch Tensorboard 사용하기 pytorch에서 tensorboard를 이용해서 log를 찍어보는 과정에 대해 알아보자! 쉽게 train log를 볼 수 있어서 나는 자주 사용하는 편이다. 일단 tensorboard를 설치해야한다. pip install tensorboard 선언 파이썬을 켜고 다음과 같이 SummaryWriter를 import한다. log_dir은 따로 명시하지 않을 시, ./runs이 default이다. log_dir은 tensorboard의 log 기록이 저장될 공간을 의미한다. from torch.utils.tensorboard import SummaryWriter log_dir = "./log_dir" writer = SummaryWriter(log_dir) 기록 사전 준비는 끝났다. 이제 기록할 값을 넣어주.. 2021. 1. 13. 이전 1 다음
