tensorflow一次运行一批

Question

我通过shuffle_batch输入管道加载我的数据。但是，当我去训练时，我想训练一下，然后蟒蛇一些东西，然后继续训练。但是，我不确定如何从shuffle_batch阅读器和文件名队列中获取对该事物的控制权。它只是继续阅读和阅读...

编辑:: 我意识到这是＆＃34;旧方式＆＃34;导入数据。但是，我并没有立即找到一种方法来解决这个问题，并采用了新的方式来解决这个问题。 https://www.tensorflow.org/versions/master/api_docs/python/tf/data/FixedLengthRecordDataset

如何通过我的培训管道只提供50条cifar记录，然后在我的jupyter笔记本中恢复控制？

Answer 1

基于你使用https://github.com/tensorflow/models/blob/master/tutorials/image/cifar10/ - 实际的训练是executing the train_op时发生的。你应该能够把你的逻辑放在那里。 e.g：

while not mon_sess.should_stop():
  mon_sess.run(train_op)
  if mon_sess.run(global_step) % 10 == 0:
    # do something special

否则它还支持max_steps参数，这可能类似于您尝试限制输入。但如果你想继续接受培训，那就不会那么有用了。