tensorboard 记录 loss

trian on batch 使用 tensorboard：

for epoch in range(epochs): 之前定义记录文件的地址
在训练前：

### discriminator loss ###
disc_log_dir = 'logs'
disc_summary_writer = tf.summary.create_file_writer(disc_log_dir)

以 cyclegan 为例，训练每个epoch得到 D loss, G loss，recon , adv
在训练中记录：

### for tensorboard ###
with disc_summary_writer.as_default():
    tf.summary.scalar('D loss', d_loss[0], step=epoch)
    tf.summary.scalar('重建损失', np.mean(g_loss[3:5]), step=epoch)
    tf.summary.scalar('一致性损失', np.mean(g_loss[5:6]), step=epoch)
    tf.summary.scalar('G loss', g_loss[0], step=epoch)

---

打开 tensorboard：

cmd 进入训练程序 .py 所在目录
再 tensorboard --logdir=logs --host=127.0.0.1 ，其中logs为训练程序目录下logs文件地址

打开网址

---