tensorboard 总是将历史events文件数据一并输出

最新推荐文章于 2024-05-03 10:04:38 发布

呀呀丫丫鸭鸭

最新推荐文章于 2024-05-03 10:04:38 发布

阅读量264

点赞数

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/qq_19321559/article/details/127693073

版权

我们可以将每次运行数据放在不同文件夹中，例如运用时间戳自动生成不同的文件夹存储events：

from datetime import datetime
TIMES = "{0:%Y-%m-%d---%H-%M-%S/}".format(datetime.now())

train_logs = 'train/' + TIMES
test_logs= 'test/'   + TIMES
writer = SummaryWriter(logdir='logs')
...
writer.add_scalars(train_logs,{'ofitloss':loss_ofit.item(),'droploss':loss_drop.item()}, epoch)
writer.add_scalars(test_logs,{'droploss-test':drop_loss.item(),'origloss-test':orig_loss.item()}, epoch)

效果如下：

在这里插入图片描述