如何读取tensorboard文件

from tensorboardX import SummaryWriter
from tensorboard.backend.event_processing import event_accumulator

file_name = 'events.out.tfevents.1631881842.f991171001781011ec08b01045e6de32e845-chend2498-0'

ea = event_accumulator.EventAccumulator(file_name)

ea.Reload()
print(ea.Tags())

print(ea.Scalars('tutankham/reward'))

writer = SummaryWriter()

for event in ea.Scalars('tutankham/reward'):
    if event.step > 10000000:
        break
        
    writer.add_scalar('tutankham/reward', event.value, event.step)

writer.close()

 

posted on 2021-09-28 19:45  穷酸秀才大草包  阅读(434)  评论(0)    收藏  举报

导航