AI/Pytorch (3) 썸네일형 리스트형 Pytorch visualization #2. TensorboardX 사용법 파이토치를 이용하다가 실시간 visualization을 하기 위해여 visdom을 사용하였다.하지만......쓰다쓰다 도저히 엉성하고 어설퍼서 tensorboard를 사용하기로 했다. 그리고 내린 결론은 역시!!!! Tensorboard를 써야겠다!! 는 것이다.사실 실시간으로 visualize 해야될 필요는 크게 없다. 어차피 그래프 plot 시키게 짜놨으니까.근데 왜 쓸데없이 계속 이작업을 하고있는지 모르겠다.본 글에서는 TensorboardX의 설치와 수행방법, 그리고 변수, 이미지, text, pr_curve, json export 등을 다룬다. Pytorch TensorboardX Tutorial- 본 블로그 실습코드 : https://github.com/keep-steady/etc_code/b.. Pytorch 사용법 정리 1. 모델 Save & Load 1) 모델 선언 model = Net() 2) 모델 저장 torch.save(model.state_dict(), '/model/model.pth') - 모델 파라미터들과 저장 경로 설정 3) 모델 로드 model = Net() model.load_state_dict(torch.load('/model/model.pth')) - 모델 선언 후, 저정된 모델파일을 불러온다 Pytorch visualization #1. Visodom 사용법 파이토치를 이용하다가 실시간 visualization을 하기 위해여 visdom을 사용하였다. Tensorboad를 사용하다가 visdom을 써보니 편한듯 하지만 좀 어설퍼보인다는 생각이 자꾸 든다. 최근호님의 튜토리얼을 현재 버전에 맞게 수정하고 그림을 추가하였다. Visdom Tutorial 튜토리얼, 현재 버전에 맞게 변형 및 그림 추가 - 본 블로그 실습 코드 : https://github.com/keep-steady/etc_code/blob/master/visdom_tutorial.ipynb - 공식 사이트 : https://github.com/facebookresearch/visdom - 튜토리얼 : https://github.com/GunhoChoi/PyTorch-FastCampus/blob.. 이전 1 다음
