加载日志数据
from tensorboard.backend.event_processing import event_accumulator
#加载日志数据
# .v2后缀的文件就是所需的日志文件
ea=event_accumulator.EventAccumulator('./callbacks/train/events.out.tfevents.1595652407.DESKTOP-73O3HSV.19208.58753.v2')
ea.Reload()
1
ea.Tags() # 查看数据文件中的数据标签
loss = ea.scalars.Items('epoch_loss') # 查看指定的标量数据
loss
2
print([(i.step,i.value) for i in loss]) # 获取每个epoch对应的loss
3
网友评论