pytorch代码中结合tensorboardX理解-20200609

最新推荐文章于 2024-08-09 07:26:21 发布

xiekai1116

最新推荐文章于 2024-08-09 07:26:21 发布

阅读量249

点赞数

分类专栏： PyTorch

本文链接：https://blog.csdn.net/csdnxiekai/article/details/106636411

版权

PyTorch 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

联系pytorch代码中结合tensorboardX进行显示, 但writer.export_scalars_to_json("./test.json")中test.json为{},进行了一下搜索

pytorch: PyTorch中使用 Tensorboard
下列代码中输出是对应data/scalar_group中的x,y.loss,前面的writer.add_scalar(‘data/x’, x, epoch)并没有在test.json输出.
test.json中内容为{“runs/Jun09_10-37-56_czey/data/scalar_group/x”: [[1591670276.9055848, 0, 66.66666412353516]
1591670276.9055848为系统时间,0为epoch,最后的66.66666412353516即为x值
Got it!
https://github.com/lanpa/tensorboardX/issues/196提示scalars

import torch
from tensorboardX import SummaryWriter

writer = SummaryWriter()
x = torch.FloatTensor([100])
y = torch.FloatTensor([500])

for epoch in range(100):
    x /= 1.5
    y /= 1.5
    loss = y - x
    print(loss)
    writer.add_histogram('zz/x', x, epoch)
    writer.add_histogram('zz/y', y, epoch)
    writer.add_scalar('data/x', x, epoch)
    writer.add_scalar('data/y', y, epoch)
    writer.add_scalar('data/loss', loss, epoch)
    writer.add_scalars('data/scalar_group', {'x': x,
                                             'y': y,
                                             'loss': loss}, epoch)
    writer.add_text('zz/text', 'zz: this is epoch ' + str(epoch), epoch)
# export scalar data to JSON for external processing
writer.export_scalars_to_json("./test.json")
writer.close()