首页 > 其他 > 详细

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

时间:2019-03-10 20:48:10      阅读:124      评论:0      收藏:0      [点我收藏+]

在一个计算图中,可以通过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 所有计算滑动平均值的变量 计算变量的滑动平均值

 

 

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

原文:https://www.cnblogs.com/lyh-vip/p/10506877.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!