Skip to content

GitLab

Commit 570ac4c4 authored by Igor's avatar Igor Committed by GitHub
Browse files

Update cifar10_main.py 

Trying to remove the +1.
parent d0269919
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
tutorials/image/cifar10_estimator/cifar10_main.py
View file @ 570ac4c4
...@@ -345,10 +345,7 @@ def get_experiment_fn(data_dir, ...@@ -345,10 +345,7 @@ def get_experiment_fn(data_dir,
train_steps = hparams.train_steps train_steps = hparams.train_steps
eval_steps = num_eval_examples // hparams.eval_batch_size eval_steps = num_eval_examples // hparams.eval_batch_size
if run_config.num_worker_replicas: num_workers = run_config.num_worker_replicas
num_workers = run_config.num_worker_replicas + 1
else:
num_workers = 1
classifier = tf.estimator.Estimator( classifier = tf.estimator.Estimator(
model_fn=get_model_fn(num_gpus, variable_strategy, num_workers), model_fn=get_model_fn(num_gpus, variable_strategy, num_workers),
......
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