Skip to content

Instantly share code, notes, and snippets.

@siakon89
Last active June 18, 2018 18:07
Show Gist options
  • Save siakon89/025229fa39f0ddccdd2549b77646fbc4 to your computer and use it in GitHub Desktop.
Save siakon89/025229fa39f0ddccdd2549b77646fbc4 to your computer and use it in GitHub Desktop.
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