tf.train.shuffle_batch

2018/1/13镜像同步10 回复

使用tf.train.shuffle_batch 取出tfrecords里的数据。训练数据有3200个，batch size 为32，训练100个batch以后就取不了数据了。求问，怎么循环多次上面训练100batch这个过程，比如说循环1000次。求讨论。

订阅后，新回复会通过你的通知中心匿名送达。

9 条回复

a543151514机器人#1 · 2018/1/14

有一个允许重复取用的开关打开就好了忘了叫啥了

tychenjiajun机器人#2 · 2018/1/14

https://tensorflow.google.cn/api_docs/python/tf/data/Dataset#repeat

zf1618701机器人#3 · 2018/1/14

allow_smaller_final_batch=True

buptwds机器人#4 · 2018/1/14

epochs？

YeShao机器人#5 · 2018/1/14

tf.train.string_input_producer([...], num_epochs=None, shuffle=True) 设置 num_epochs=None 就可以取多次了。我在tf1.2.1 版本中的例子：https://github.com/yongyehuang/Tensorflow-Tutorial/tree/master/python/the_use_of_tfrecord 新版本的 Data API 还没试过。

dsfour机器人#6 · 2018/1/14

tf.train.shuffle_batch应该不能控制能不能取到多个epoch的数据，只是随机取多个训练样本组成一个batch。数据的预处理可以参考inceptionv3的预处理过程： https://github.com/tensorflow/models/blob/master/research/inception/inception/image_processing.py

guo1994机器人#7 · 2018/1/14

epochs控制样本训练轮次

FuckerY机器人#8 · 2018/1/14

谢谢大家[ema11]，我试试[ema11]

Viredery机器人#9 · 2018/1/14

原来有函数啊，我每次都是手写的。。。。