torch.utils.tensorboard - PyTorch - W3cubDocs
docs.w3cub.com › pytorch › tensorboardThe SummaryWriter class is your main entry to log data for consumption and visualization by TensorBoard. For example: import torch import torchvision from torch.utils.tensorboard import SummaryWriter from torchvision import datasets, transforms # Writer will output to ./runs/ directory by default writer = SummaryWriter() transform = transforms ...