通过SummaryWriter记录

定义一个记录器：
tb_writer = SummaryWriter(config['tb_writer_dir'])
添加变量：
tb_writer.add_scalar('loss', np.mean(losses), int(itr_global /1000))
要动态查看代码的情况：
tensorboard --logdir=logs
即可点击连接查看。