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项。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐