텐서보드 썸네일형 리스트형 [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 다음
