take函数的问题
来源：4-7 生成tfrecords文件
wxz123
2019-11-09
对于这里的dataset.take(steps_per_shard)，意思不应该是每次都取dataset前steps_per_shard个batch吗，如果这样写的话那怎么取到steps_per_shard以后的batch呢？假如dataset有100个batch，steps_per_shard=10，这么写的的话不就只对前10个batch进行写入tfrecord文件，后90个一直也没用上吧
写回答
2回答
-
正十七
2019-11-17
同学你好，经过验证，这个API确实是我用错了，感谢你的指出。稍后我会更新代码修复这个错误。
10 -
正十七
2019-12-01
同学你好，这块的代码已经改正并传到代码库，新代码中先用了skip把前面的batch跳过，然后再take，这样可以避免这个问题。
00
相似问题