码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow中的Supervisor

时间:2018-07-31 17:18:02      阅读:160      评论:0      收藏:0      [点我收藏+]

标签:super   bsp   man   range   put   sign   compute   check   for   

  tf.train.Supervisor()可以帮我们简化一些事情,可以保存模型参数和Summary,它有以下的作用:

  1)自动去checkpoint加载数据或初始化数据 ,因此我们就不需要手动初始化或者从checkpoint中加载数据

  2)自身有一个Saver,可以用来保存checkpoint,因此不需要创建Saver,直接使用Supervisor里的Saver即可

  3)有一个summary_computed用来保存Summary,因此不需要创建summary_writer

  因此从这里看,Supervisor相当于将一些功能集成到一起了,具体代码实现

import tensorflow as tf


a = tf.Variable(1)
b = tf.Variable(2)
c = tf.add(a, b)
update = tf.assign(a, c)
tf.summary.scalar("a", a)
init_op = tf.initialize_all_variables()
merged_summary_op = tf.summary.merge_all()
sv = tf.train.Supervisor(logdir="./tmp/", init_op=init_op)  # 直接初始化
saver = sv.saver
with sv.managed_session() as sess:
    for i in range(10000):
        update_ = sess.run(update)
        if i % 10 == 0:
            merged_summary = sess.run(merged_summary_op)
            sv.summary_computed(sess, merged_summary, global_step=i)  # 直接将summary保存

        if i % 100 == 0:
            saver.save(sess, save_path="./tmp/", global_step=i)  # 直接将模型参数保存

 

tensorflow中的Supervisor

标签:super   bsp   man   range   put   sign   compute   check   for   

原文地址:https://www.cnblogs.com/jiangxinyang/p/9396211.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!