Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
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")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment