tensorboardX Deep Learning

tensorboardX(텐서보드 엑스) : Pytorch framework에서 사용 가능한 딥러닝 visualization tool



1.

Tensorflow framework에는 tensorboard라고 하는 visualization tool이 있다.
필요한 정보를 학습할때마다 log 형태로 저장하고, 이를 web기반으로 실시간 또는 학습이 끝난 뒤 확인이 가능하다.

2.

Pytorch framework의 경우 tensorflow의 tensorboard에 대응되는 visualization tool이 별도로 존재하지 않는다.
많은 유저들이 이미 익숙해져 있는 tensorflow의 tensorboard를 그대로 이용하고자 하는 것이 tensorboardX 다.



tensorboardX로 저장한 log를 tensorboard로 띄운 결과.


3. 

설치방법은 pip을 사용


pip install tensorboardX

4.

이후 실행방법은 tensorboard와 같다. (tensorboardX로 저장한 log를 tensorboard로 읽는 것이기 때문에)
tensorboard --logdir=(SummaryWriter에서 설정해놓은 logdir)


Tensorflow가 설치되어 있지 않기 때문에 tensorflow를 찾을 수 없다는 문구가 뜨는데 tensorboard사용에는 문제가 없다.

별다른 설정을 하지 않았다면 tensorboard의 포트값은 6006이다.
사용하는 web browser에서 다음의 주소로 들어가면 된다.

localhost:6006


5.


작성하다 보니 pytorch.utils 모듈 내에 tensorboard가 제공되고 있는 것을 확인하였다.



추후 수정될 수 있으며 아직 add_graph가 작업 중 이라고 한다.
pytorch 사용자들 내에서도 시각화에 대한 니즈가 있고, 이를 반영하여 대체제로 tensorboardX가 널리 사용되었던 것인 만큼
조만간 이 모듈을 통해 pytorch 자체에서 시각화가 용이해질 것으로 생각된다.


덧글

댓글 입력 영역