epoch设置过小导致OutOfRangeError

在通过读取TFRecod文件,形成batch数据训练时,为什么提示如下错误 OutOfRangeError (see above for traceback): RandomShuffleQueue ‘_2_shuffle_batch/random_shuffle_queue’ is closed and has insufficient elements (requested 2, current size 0

image_batch, label_batch = tf.train.shuffle_batch([images, labels], 
			                                              batch_size = batchsize, 
			                                              capacity = capacity, 
			                                              min_after_dequeue = min_after_dequeue)

错误提示: OutOfRangeError (see above for traceback): RandomShuffleQueue ‘_2_shuffle_batch/random_shuffle_queue’ is closed and has insufficient elements (requested 2, current size 0) [Node: shuffle_batch = QueueDequeueManyV2[component_types=[DT_FLOAT, DT_INT32], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"]] [[Node: shuffle_batch/_17 = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name=“edge_20_shuffle_batch”, tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"]]

这个问题的原因是:我只有一个TFRecord数据文件,其中样本数仅1500个。并且因为epoch设置过小,在开始训练前就把数据读完退出了。可以加大样本数量,或者把epoch设置的大一些,如果设置成Nnoe,程序会无限制地一直跑下去,当然你可以在结果足够好的时候手动中断程序的运行。
下面一行代码中,原num_epochs = 3 , 出现上述错误。将num_epochs 改为 10 即可正常运行。没有测试最小可设置多少。
不过问题是如何计算出准确的 num_epochs 值呢?

filename_queue = tf.train.string_input_producer([tfr_filename], shuffle = False, num_epochs = 10)
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值