Customize Your Data Genrators for Faster Training

Nilesh
Towards Data Science
2 min readFeb 7, 2018

--

The idea behind using a Keras generator is to get batches of input and corresponding output on the fly during training process, e.g. reading in 100 images, getting corresponding 100 label vectors and then feeding this set to the gpu for training step.

Keras : keras.io

--

--