Update cifar10_main.py

28d37e7a · Igor · GitHub · 570ac4c4 · 28d37e7a
Commit 28d37e7a authored Sep 15, 2017 by Igor Committed by GitHub Sep 15, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

tutorials/image/cifar10_estimator/cifar10_main.py tutorials/image/cifar10_estimator/cifar10_main.py +3 -4

No files found.
--- a/tutorials/image/cifar10_estimator/cifar10_main.py
+++ b/tutorials/image/cifar10_estimator/cifar10_main.py
@@ -344,11 +344,10 @@ def get_experiment_fn(data_dir,

    train_steps = hparams.train_steps
    eval_steps = num_eval_examples // hparams.eval_batch_size
-
-    num_workers = run_config.num_worker_replicas
-    
+ 
    classifier = tf.estimator.Estimator(
-        model_fn=get_model_fn(num_gpus, variable_strategy, num_workers),
+        model_fn=get_model_fn(num_gpus, variable_strategy,
+                              run_config.num_worker_replicas or 1),
        config=run_config,
        params=hparams)