[PyTorch] 학습 Loss 그래프 그리기 (Tensorboard 이용)
모델 학습 과정에서의 Loss를 그래프로 시각화하기 위해, Tensorboard를 이용하는 방법을 알아본다. 참조1 : seongkyun.github.io/others/2019/05/11/pytorch_tensorboard/ 참조2 : sensibilityit.tistory.com/512 참조한 블로그들이다. 이 포스팅에서는 핵심만 간략하게 설명할 것이니 자세한 설명이 필요하다면 위 블로그에서 확인하면 된다 :) 1. Tensorboard 설치하기 !pip install tensorboardX tensorflow도 설치되어 있어야한다고 한다. 2. 모듈 임포트 및 객체 생성 from tensorboardX import SummaryWriter summary = SummaryWriter() 3. loss ..
AI
2021. 5. 11. 16:56