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

编程入门行业动态更新时间:2024-10-13 02:19:51

【pytorch】利用tensorboard记录训练<a href=https://www.elefans.com/category/jswz/34/1769466.html style= 过程与查看"/>

代码

from torch.utils.tensorboard import SummaryWriter

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

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

命令行输入：

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

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

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

更多推荐

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

本文发布于:2023-07-28 21:39:20，感谢您对本站的认可！

过程 pytorch tensorboard

评论列表（有 0 条评论）