有时候需要修改数据集大小,使用 paddlepaddle 如何修改呢?
paddlepaddle 官方提供的shuffle方法可以做到:
paddle.reader.shuffle(reader, buf_size)
Parameters:
- reader (callable) – the original reader whose output will be shuffled.
- buf_size (int) – shuffle buffer size.
其中,buf_size即可以调整你所需要灌入模型的输入数据集大小。
参考 :paddlepaddle paddle.reader.shuffle()