Skip to content

编程细节相关

人工智能

怎样才能在tensorboard中把多个loss的曲线画到同一张图上,便于对比?

用两个路径不同的wirter,在两个不同的文件夹下,把参数名称写成同一个。

writer_g = SummaryWriter("runs/exp1/generator/")
writer_d = SummaryWriter("runs/exp1/discriminator/")
writer_g.add_scalar('loss', loss_g, iteration)
writer_d.add_scalar('loss', loss_d, iteration)