Skip to content

Instantly share code, notes, and snippets.

@hadifar
Created January 5, 2019 15:40
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 hadifar/339c00de3f9ba300f8b2aac4020f9d43 to your computer and use it in GitHub Desktop.
Save hadifar/339c00de3f9ba300f8b2aac4020f9d43 to your computer and use it in GitHub Desktop.
# Instantiate a Keras inception v3 model.
keras_inception_v3 = tf.keras.applications.inception_v3.InceptionV3(weights=None)
keras_inception_v3.compile(optimizer=tf.keras.optimizers.SGD(lr=0.0001, momentum=0.9),
loss='categorical_crossentropy',
metric='accuracy')
est_inception_v3 = tf.keras.estimator.model_to_estimator(keras_model=keras_inception_v3)
train_input_fn = tf.estimator.inputs.numpy_input_fn(
x={"input_1": train_data},
y=train_labels,
num_epochs=1,
shuffle=False)
est_inception_v3.train(input_fn=train_input_fn, steps=2000)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment