tensorboard / tensorboardX 无法记录超过1000个点

问题描述

tensorboard / tensorboardX 记录的数据不是连续的,即使在代码中是连续记录

最小复现代码(tensorboardX)

from tensorboardX import SummaryWriter
writer = SummaryWriter('test')
for i in range(2000):
    writer.add_scalar('t/test_value',scalar_value=i,global_step=i)
print('finished')

运行最小复现代码,可以发现,使用tensorboard命令启动的WEBUI,只截取了1000个值,两个sample之间的间隔并不是连续的。
无论是左侧的webui还是右侧导出的csv表,其step都不是连续的

解决方法

参考stackoverflow page

tensorboardX 下

添加 --samples_per_plugin scalars=999999999参数,将总的采样点数设置成一个非常大的值,全部命令如下(由于我在服务器上运行,还有个--host,忽略就好):

$ tensorboard --host [YOURIP] --logdir test --samples_per_plugin scalars=999999999

tensorboard下

可以找到tensorboard包安装目录,一般在[YOUR PYTHON ENV]/site-packages/tensorflow/tensorboard/backend里。
打开application.py,修改scalar_metadata.PLUGIN_NAME

解决结果

导出的CSV已经可以每个STEP都记录

  • 5
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值