from torch.utils.tensorboard import SummaryWriter
write = SummaryWriter(log_dir='log_dir',flush_secs=1)
write.add_scalar(tag='loss',scalar_value=loss,global_step=epoch*batch_num+iteration+1)
write.add_graph(net,images)
write.close()
tensorboard --logdir=E:/py-pro/liu-runda-yolov3-train-master/train2/train/log_dir/ --host=127.0.0.1