这是一个非常粗糙的记录,如果你偶然进入这个页面,那么,请注意!如果没看懂,请立刻放弃,并寻找更好的例子
import tensorflow as tf
a = tf.Variable(1)
b = tf.Variable(2)
c = tf.add(a, b)
saver = tf.train.Saver()
saver_hook = tf.train.CheckpointSaverHook('./testpot/',
save_steps=2,
saver=saver)
global_step = tf.Variable(0, name='global_step', trainable=False)
summary_op = tf.summary.scalar('c', c)
summary_hook = tf.train.SummarySaverHook(save_steps=2,
summary_op=summary_op)
with tf.train.MonitoredTrainingSession(hooks=[saver_hook, summary_hook]) as sess:
while not sess.should_stop():
print(sess.run(c))
说明
tf.train.Supervisor
已经被放弃,github上建议使用MonitoredTrainingSession
替代Supervisor
在使用MonitoredTrainingSession
前,需要有
1. saver
2. saver_hook
3. global_step
4. summary_op
5. summary_hook
这些变量,才能保证程序不报错。
附
这个程序用 Tensorboard 打开看到的图
不错
请问如何将tf.train.Supervisor转为MonitoredTrainingSession呢
不好意思,这方面我无法提供帮助。
我是通过重写代码实现从
Supervisor
转到MonitoredTrainingSession
的。