from torch.utils.tensorboard import SummaryWriter # Creating a summary writer writer = SummaryWriter() # Writing scalar data to TensorBoard writer.add_scalar('training/accuracy', 0.98, 1) writer.add_scalar('training/loss', 0.05, 1)
from torch.utils.tensorboard import SummaryWriter # Creating a summary writer writer = SummaryWriter() # Writing scalar data to TensorBoard writer.add_scalar('training/learning_rate', 0.0001, 1)This module belongs to the PyTorch library, and specifically the torch.utils.tensorboard package.