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