Update cifar10_main.py

Trying to remove the +1.

Update cifar10_main.py
Trying to remove the +1.
570ac4c4 · Igor · GitHub · d0269919 · 570ac4c4
Commit 570ac4c4 authored Sep 15, 2017 by Igor Committed by GitHub Sep 15, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 5 deletions

tutorials/image/cifar10_estimator/cifar10_main.py tutorials/image/cifar10_estimator/cifar10_main.py +2 -5

No files found.
--- a/tutorials/image/cifar10_estimator/cifar10_main.py
+++ b/tutorials/image/cifar10_estimator/cifar10_main.py
@@ -345,11 +345,8 @@ def get_experiment_fn(data_dir,
    train_steps = hparams.train_steps
    eval_steps = num_eval_examples // hparams.eval_batch_size

-    if run_config.num_worker_replicas:
-      num_workers = run_config.num_worker_replicas + 1
-    else:
-      num_workers = 1
-
+    num_workers = run_config.num_worker_replicas
+    
    classifier = tf.estimator.Estimator(
        model_fn=get_model_fn(num_gpus, variable_strategy, num_workers),
        config=run_config,