Tensorboard 썸네일형 리스트형 torch에서 Tensorboard를 이용해서 loss plot 하는 방법 torch에서 Tensorboard를 이용해서 loss plot 하는 방법 사용자 입장에서 파이토치 라이브러리를 사용하면서 필요한 부분에 대하여 포스팅 한 글 입니다. Plot training loss with tensorboard 1. 파이썬 코드에서 SummaryWriter를 import 해줍니다. from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter() 2. Training에서 Loss를 계산하고 아래 코드를 이용해서 로컬로 파일을 써줍니다. 아마 다른 코드가 궁금할 수 있는데, writer를 다른 코드와 dependency가 없이 값을 써주는 것 입니다. 따라서 원하는 위치에 적어주시면 됩니다. writer.add_sca.. 더보기 이전 1 다음
