【Tensorflow】TFRecord数据的写入和读取经验
问题描述:利用tensorflow进行神经网络训练,当数据集较小或者内存足够大时,通常的做法是将全部数据集加载到内存里,然后再将数据集分批feed给网络进行训练(一般配合yield使用效果更佳)。但是,当数据集大到内存不足以全部加载进来的时候,必须寻找新的加载数据的方法。 解决办法: 可以尝试使用tensorflow提供的队列queue,训练时从文件中分批读取数据。这里选择tensorflwo...
原创
2018-08-27 17:05:40 ·
5568 阅读 ·
4 评论