Tensorboard 썸네일형 리스트형 [Pytorch] 텐서보드 초간단 사용법 SummaryWriter 임포트 from torch.utils.tensorboard import SummaryWriter Writer 생성 writer = SummaryWriter() 위와 같이 writer을 생성하면 기본값으로 ./runs에 로그가 저장된다. 저장경로를 바꾸고 싶으면 다음과 같이 하면된다. writer = SummaryWriter(log_dir) 로그 찍을 변수 넣어주기 writer.add_scalar('Loss/train', train_loss, epoch) 원하는 이름을 적고, 해당 이름에 대한 값을 전달해준 뒤에, 어떤 step에 대해서 로그를 찍을 것인지 epoch를 전달해주었다. Writer 닫기 writer.close() 텐서보드 실행 tensorboard --logdir=r.. 더보기 원격 서버 도커에서 텐서보드 사용하는 법 큰 그림부터 보면 다음과 같다. 1. Docker run 할때 포트 열어주기 2. 서버에서 텐서보드 열기 3. 로컬 컴퓨터에서 ssh로 서버와 포트 연결 4. 로컬 컴퓨터에서 url로 접속 1. 서버에서 Docker run 할 때 포트 열어주기 docker run -it\ -p 6008:6008 \ # 사용할 포트번호. 이전 것과 겹치지 않아야 한다. --runtime=nvidia \ --shm-size 32G \ -e NVIDIA_VISIBLE_DEVICES=0 \ # 사용할 gpu 인덱스 --name=name\ # 컨테이너 이름 -v /data/codes:/mycodes \ # 서버 디렉토리:컨테이너 디렉토리 연결 pytorch:latest\ # 이미지 /bin/bash 2. 서버에서 tensorbo.. 더보기 이전 1 다음
