Tensor Flow shuffle_batch() blocks at end of epoch

前端 未结 3 1555
醉话见心
醉话见心 2021-01-23 00:57

I\'m using tf.train.shuffle_batch() to create batches of input images. It includes a min_after_dequeue parameter that makes sure there\'s a specified number of elements inside t

3条回答
  •  没有蜡笔的小新
    2021-01-23 02:00

    There is an optional argument allow_smaller_final_batch

    "allow_smaller_final_batch: (Optional) Boolean. If True, allow the final batch to be smaller if there are insufficient items left in the queue."

提交回复
热议问题