Skip to content

GitLab

Commit 28d37e7a authored by Igor's avatar Igor Committed by GitHub
Browse files

Update cifar10_main.py

parent 570ac4c4
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
tutorials/image/cifar10_estimator/cifar10_main.py
View file @ 28d37e7a
...@@ -345,10 +345,9 @@ def get_experiment_fn(data_dir, ...@@ -345,10 +345,9 @@ 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
num_workers = run_config.num_worker_replicas
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,
run_config.num_worker_replicas or 1),
config=run_config, config=run_config,
params=hparams) params=hparams)
......
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