Hello Computer Vision

torch SummaryWriter 에 관한 자그만한 정보 본문

딥러닝/파이토치

torch SummaryWriter 에 관한 자그만한 정보

지웅쓰 2024. 1. 5. 17:14

IDE는 Pycharm을 사용했고 기존에는 항상 Google Colab에서 list에다 결과값을 저장하고 matplotlib을 사용해서 시각화했는데 이제 tensorboard를 쓸 필요성을 느껴서 이번에 처음 써보았다.

 

우선 pip install tensorboarrd를 통해 tensorboard를 수행하면은 나만 그런지 모르겠는데

WARNING:tensorflow:From C:\Users ..... tf.compat.v1.losses.sparse_softmax_cross_entropy instead.
이러한 에러가 뜬다. 근데 결과적으로는 아무 문제없다.

 

보통 모든 설명글에서는 저장한 값에 대하여 tensorboard를 열고 싶으면 tensorboard --logdir=runs를 terminal에 입력하라고 한다. 그 이유는 SummaryWriter 객체를 생성하고 기본 root값이 runs이기 때문이다. 그러나 나는 이렇게 수행할 경우 tensorboard에서 이미지가 나오지않았다. 우선 내 프로젝트 디렉터리의 구조를 한번 보여주자면

현재 FixMatch안의 py파일을 통해 scalaer 값들을 tensorboard에 저장했다. 따라서 내 값들이 저장된 runs 디렉토리의 주소는 FixMatch/runs 인 것이다(물론 주소 자체를 FixMatch로 옮기면은 runs만 해도 될 것이다). 따라서 나와 같은 구조를 가진 사람들은 

tensorboard --logdir=(상위주소)/runs 를 수행하면 될 것이다.