Skip to content

GitLab

Commit 308c7934 authored by Toby Boyd's avatar Toby Boyd Committed by guptapriya
Browse files

update v2 check and fix ncf v2 check error logic.

parent 6f47c378
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/ncf_keras_main.py
View file @ 308c7934
...@@ -275,7 +275,7 @@ def run_ncf(_): ...@@ -275,7 +275,7 @@ def run_ncf(_):
num_gpus=FLAGS.num_gpus) num_gpus=FLAGS.num_gpus)
params["distribute_strategy"] = strategy params["distribute_strategy"] = strategy
if keras_utils.is_v2_0() and strategy is not None: if not keras_utils.is_v2_0() and strategy is not None:
logging.error("NCF Keras only works with distribution strategy in TF 2.0") logging.error("NCF Keras only works with distribution strategy in TF 2.0")
return return
......
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