Skip to content

Instantly share code, notes, and snippets.

@obeshor

obeshor/optimizer.py

Last active June 25, 2019 22: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 obeshor/bb1dfc8627ff8e50563084bb30111481 to your computer and use it in GitHub Desktop.
Save obeshor/bb1dfc8627ff8e50563084bb30111481 to your computer and use it in GitHub Desktop.
Download ZIP
Adam Optimizer
Raw
optimizer.py
|#Compile model specifying the optimizer learning rate
LEARNING_RATE = 0.001 #@param {type:"number"}
model.compile(
optimizer=tf.keras.optimizers.Adam(lr=LEARNING_RATE),
loss='categorical_crossentropy',
metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment