Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "7d0a47f387e7c76ffa4fee5e7365228cef25801d"
Commit 28f6182f authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Estimator should still use v1 optimizer. 

PiperOrigin-RevId: 301308906
parent 69837027
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/neumf_model.py
View file @ 28f6182f
...@@ -109,7 +109,7 @@ def neumf_model_fn(features, labels, mode, params): ...@@ -109,7 +109,7 @@ def neumf_model_fn(features, labels, mode, params):
mlperf_helper.ncf_print(key=mlperf_helper.TAGS.OPT_HP_ADAM_EPSILON, mlperf_helper.ncf_print(key=mlperf_helper.TAGS.OPT_HP_ADAM_EPSILON,
value=params["epsilon"]) value=params["epsilon"])
optimizer = tf.keras.optimizers.Adam( optimizer = tf.compat.v1.train.AdamOptimizer(
learning_rate=params["learning_rate"], learning_rate=params["learning_rate"],
beta1=params["beta1"], beta1=params["beta1"],
beta2=params["beta2"], beta2=params["beta2"],
......
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