Hello i’m newbie in pytorch and vision.
I have a code like this.
from engine import train_one_epoch # Training writer = SummaryWriter() start = datetime.now() for epoch in range(epochs): # train for one epoch, printing every 50 iterations train_one_epoch(model, optimizer, loader_train, device, epoch, print_freq=20) # update the learning rate lr_scheduler.step() # evaluate on the test dataset evaluate(model, loader_val, device=device) end = datetime.now()
Anyone can tell me how to add a tensorboard to monitor my training. Thanks