Skip to content

GitLab

Unverified Commit 9057a3a1 authored by Yanhui Liang's avatar Yanhui Liang Committed by GitHub
Browse files

Update optimizer (#4491)

parent 9bf9a837
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/recommendation/ncf_main.py
View file @ 9057a3a1
...@@ -146,8 +146,7 @@ def convert_keras_to_estimator(keras_model, num_gpus, model_dir): ...@@ -146,8 +146,7 @@ def convert_keras_to_estimator(keras_model, num_gpus, model_dir):
Returns: Returns:
est_model: The converted Estimator. est_model: The converted Estimator.
""" """
# TODO(b/79866338): update GradientDescentOptimizer with AdamOptimizer optimizer = tf.train.AdamOptimizer(
optimizer = tf.train.GradientDescentOptimizer(
learning_rate=FLAGS.learning_rate) learning_rate=FLAGS.learning_rate)
keras_model.compile(optimizer=optimizer, loss="binary_crossentropy") keras_model.compile(optimizer=optimizer, loss="binary_crossentropy")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment