wandb1 Tensorboard and WandB Tensorboard와 WandB는 머신러닝과 딥러닝 모델 개발에 대한 모니터링 툴이다. Tensorboard는 이름에서 알 수 있듯이 TensorFlow와 연계된 툴이지만 PyTorch에서도 사용할 수 있다. 여기서는 AlexNet을 이용해 CIFAR-10을 classification하는 예시를 통해 살펴보고자 한다. Tensorboard 예시 from torch.utils.tensorboard import SummaryWriter # TensorBoard 사용 준비 writer = SummaryWriter("./runs/cv/alexnet") writer를 이용하여 텐서보드에 기록할 준비를 한다. tensorboard에는 add_scalar, add_scalars, add_graph, add_imag.. 2024. 3. 29. 이전 1 다음
