update v2 check and fix ncf v2 check error logic.

308c7934 · Toby Boyd · guptapriya · 6f47c378 · 308c7934
Commit 308c7934 authored Jul 19, 2019 by Toby Boyd Committed by guptapriya Jul 19, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/recommendation/ncf_keras_main.py official/recommendation/ncf_keras_main.py +1 -1

No files found.
--- a/official/recommendation/ncf_keras_main.py
+++ b/official/recommendation/ncf_keras_main.py
@@ -275,7 +275,7 @@ def run_ncf(_):
      num_gpus=FLAGS.num_gpus)
  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")
    return