使用python代码实现tensorflow event文件内容读取
tf_event_file读取使用tensorboard可以对events.out.tfevents.进行可视化,如需读取其内loss或acc等曲线进行custom处理的话,则需使用读取函数将其数值读取出来,读取函数如下:import osimport jsonimport tensorflow as tfdef get_loss_from_tfevent_file(tfevent...
·
tf_event_file读取
使用tensorboard可以对events.out.tfevents.进行可视化,如需读取
其内loss或acc等曲线进行custom处理的话,则需使用读取函数将其数值读取出来,读取函数如下:
import os
import json
import tensorflow as tf
def get_loss_from_tfevent_file(tfevent_filename):
"""
:param tfevent_filename: the name of one tfevent file
:return: loss (list)
"""
loss_val_list = []
for event in tf.train.summary_iterator(tfevent_filename):
for value in event.summary.value:
# print(value.tag)
if value.HasField('simple_value'):
if value.tag == "loss":
# print(value.simple_value)
loss_val_list.append(value.simple_value)
return loss_val_list
函数输入为tfevent文件名,输出为loss取值列表。注意如需读取其他变量值,可以修改函数中value.tag项。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)