tensorflow中检查点之间的间隔
如何在tensorflow中指定两个连续检查点之间的间隔?tensorflow中检查点之间的间隔,tensorflow,checkpoint,Tensorflow,Checkpoint,如何在tensorflow中指定两个连续检查点之间的间隔?tf.train.Saver中没有指定该选项。每次我以不同数量的全局步骤运行模型时,我都会在检查点之间获得一个新的间隔这是一个用于写入检查点的“被动”实用程序,它只在其他代码调用其方法时写入检查点。因此,检查点的写入速率取决于您用来训练模型的框架: 如果您正在使用低级TensorFlow API(tf.Session)并编写自己的训练循环,您只需在自己的代码中插入对Saver.save()的调用即可。一种常见的方法是基于迭代计数进行此操
tf.train.Saver
中没有指定该选项。每次我以不同数量的全局步骤运行模型时,我都会在检查点之间获得一个新的间隔这是一个用于写入检查点的“被动”实用程序,它只在其他代码调用其方法时写入检查点。因此,检查点的写入速率取决于您用来训练模型的框架:
- 如果您正在使用低级TensorFlow API(
)并编写自己的训练循环,您只需在自己的代码中插入对tf.Session
的调用即可。一种常见的方法是基于迭代计数进行此操作:Saver.save()
for i in range(NUM_ITERATIONS): sess.run(train_op) # ... if i % 1000 == 0: saver.save(sess, ...) # Write a checkpoint every 1000 steps.
- 如果您使用的是为您写入检查点的,则可以在构造函数中指定检查点间隔(以秒为单位)。默认情况下,它每10分钟保存一个检查点。要将此更改为“每分钟”,请执行以下操作:
with tf.train.MonitoredTrainingSession(..., save_checkpoint_secs=60): # ...
- 谢谢!
这解决了我的问题:
tf.contrib.slim.learning.train(
火车站,
检查站,
log_every_n_steps=args.log_every_n_steps,
图=g,
全局步骤=model.global步骤, 步数=args.number步数, init_fn=model.init_fn, 保存摘要\u秒=300, 保存时间间隔为300秒, 储蓄者=储蓄者)