collection管理程序中不同类别的资源

时间:2023-12-06 08:57:20

在一个计算图中,可以通过collection管理不同类别的资源,如通过tf.add_to_collection函数可以将资源加入一个或多个集合中,然后通过tf.get_collection获取一个集合里的所有资源,这里的资源可以使张量,变量或者运算tf程序所需的队列资源...

Tensorflow中维护的集合列表

集合名称 集合内容 使用场景
tf.GraphKeys.VARIABLES 所有变量 持久化TensorFlow模型
tf.GraphKeys.TRAINABLE_VARIABLES 可学习的变量(一般指神经网络中的参数) 模型训练,生成模型可视化内容
tf.GraphKeys.SUMMARIES 日志生产相关的变量 TensorFlow计算可视化
tf.GraphKeys.QUEUE_RUNNERS 处理输入的QueueRunner 输入处理
tf.GraphKeys.MOVING_AVERAGE_VARIABLES 所有计算滑动平均值的变量 计算变量的滑动平均值