epochs = 10
model.fit_generator(datagen.flow(x_train, y_train,
                                     batch_size=batch_size),
                        epochs=epochs,
                        validation_data=(x_test, y_test))

model_name='cifar-10-cnn-'+str(epochs)
model.save(model_path+'/'+model_name+'.hd5') # Keras model
print("Saved Keras model")