Не очень понятен вопрос - вот пример использования model.fit_generator:
# initialize the number of epochs and batch size
EPOCHS = 100
BS = 32
 
# construct the training image generator for data augmentation
aug = ImageDataGenerator(rotation_range=20, zoom_range=0.15,
	width_shift_range=0.2, height_shift_range=0.2, shear_range=0.15,
	horizontal_flip=True, fill_mode="nearest")
 
# train the network
H = 
model.fit_generator(aug.flow(trainX, trainY, batch_size=BS),
	validation_data=(testX, testY), steps_per_epoch=len(trainX) // BS,
	epochs=EPOCHS)
подробней в 
https://russianblogs.com/article/6460835835/