rueki

tensorboardx ssh 연결 및 모니터링 방법 본문

pytorch

tensorboardx ssh 연결 및 모니터링 방법

륵기 2023. 7. 31. 11:48
728x90
반응형

로컬에서 학습하는 것 모니터링 해당 X

리눅스 서버를 통해서 학습하는 내용 monitoring 확인 시 내 로컬에서 확인하기 위함.

ssh -NfL localhost:6006:localhost:6006 "서버 접속자 id"@"서버 주소"

맨 뒤에서 사용하는 서버의 정보에 대한 것만 개별 설정하면 끝.

localhost 번호 변경은 사용자 마음.

 

이 전에 tensorboard를 서버에서 실행시켜줘야한다.

tensorboard --logdir="텐서보드 log기록 경로" --port=6006

train 코드에서 tensorboardx를 사용하게 되면 본 폴더 내에 runs라는 기록 저장 폴더가 생긴다.

그 폴더 경로를 logdir에 넣어주면 해당 폴더 내의 실험 기록들이 전부 올라가며, 학습 중인 것도 실시간으로 확인이 가능함.

 

728x90
반응형
Comments