Commit 41d071ee authored by guptapriya's avatar guptapriya
Browse files

NCF Keras: Fail early with TF 1.x + dist strat

This combination does not yet work. Fail early with an explicit message instead of throwing error later on.
parent 97a87f9c
......@@ -275,6 +275,10 @@ def run_ncf(_):
num_gpus=FLAGS.num_gpus)
params["distribute_strategy"] = strategy
if keras_utils.is_v2_0() and strategy is not None:
logging.error("NCF Keras only works with distribution strategy in TF 2.0")
return
if (params["keras_use_ctl"] and (
not keras_utils.is_v2_0() or strategy is None)):
logging.error(
......
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