Skip to content

Instantly share code, notes, and snippets.

@siakon89

siakon89/train_cnn_cifar_10.py

Last active June 18, 2018 18:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save siakon89/025229fa39f0ddccdd2549b77646fbc4 to your computer and use it in GitHub Desktop.
Save siakon89/025229fa39f0ddccdd2549b77646fbc4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_cnn_cifar_10.py
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