Skip to content

GitLab

Commit c8be4828 authored by Taylor Robie's avatar Taylor Robie
Browse files

update call to TPUStrategy

parent 2d51c3f1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/ncf_main.py
View file @ c8be4828
...@@ -95,7 +95,7 @@ def construct_estimator(model_dir, params): ...@@ -95,7 +95,7 @@ def construct_estimator(model_dir, params):
os.environ['TF_CONFIG'] = json.dumps(tf_config_env) os.environ['TF_CONFIG'] = json.dumps(tf_config_env)
distribution = tf.contrib.distribute.TPUStrategy( distribution = tf.contrib.distribute.TPUStrategy(
tpu_cluster_resolver, 100, params["batches_per_step"]) tpu_cluster_resolver, steps_per_run=100)
else: else:
distribution = distribution_utils.get_distribution_strategy( distribution = distribution_utils.get_distribution_strategy(
......
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