Tensorboard & Wandb

Tensorbord

학습하는 중간에 데이터가 어떻게 변하는지 확인해줄 수 있도록 실시간 모니터링을 지원해주는 도구

• scalar : metrci 등 상수 값의 연속(epoch)을 표시
• graph : 모델의 computational graph 표시
• histogram : weight 등 값의 분포를 표현
• image : 예측 값과 실제 값을 비교 표시
• mesh : 3d 형태의 데이터를 표현하는 도구

writer = SummaryWriter(logs_base_dir) # SummaryWriter에 기록할 위치를 지정
writer.flush() # 기록 (따로 안정해줘도 자동으로 flush해주긴 함)

파일 기록할 때 보통 [프로젝트명(지표), 이름]순으로 파일 위치를 기록해줌

텐서보드를 띄우는 방법
%load_ext tensorboard # 주피터나 코랩에서 바로 사용
%tensorboard --logdir {logs_base_dir} # 6006 포트로 자동으로 생성

참고

모든 자료는 부스트캠프 AI Tech교육 자료를 참고