【pytorch】利用tensorboard记录训练过程与查看

文章目录

  • 代码
  • 读取保存的log


代码

from torch.utils.tensorboard import SummaryWriter

在epoch内保存,三个参数分别为保存字符串名、保存变量和步数

writer.add_scalar('training loss', running_loss / 10, epoch * size + step)

读取保存的log

命令行输入:

tensorboard --logdir /home/...(events保存的文件名)

如果把所有的log都存放在runs文件下,地址可以到runs为止,这样就可以在tensorboard查看、对比多次的实验的数据。

tensorboard --logdir /home/.../runs

你可能感兴趣的:(pytorch,pytorch)