Skip to content

GitLab

Commit 0319908c authored by Zhichao Lu's avatar Zhichao Lu Committed by lzc5123016
Browse files

Fix typos to enable sync training. 

PiperOrigin-RevId: 188251743
parent e55685d7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/trainer.py
View file @ 0319908c
...@@ -258,10 +258,10 @@ def train(create_tensor_dict_fn, create_model_fn, train_config, master, task, ...@@ -258,10 +258,10 @@ def train(create_tensor_dict_fn, create_model_fn, train_config, master, task,
sync_optimizer = None sync_optimizer = None
if train_config.sync_replicas: if train_config.sync_replicas:
training_optimizer = tf.SyncReplicasOptimizer( training_optimizer = tf.train.SyncReplicasOptimizer(
training_optimizer, training_optimizer,
replicas_to_aggregate=train_config.replicas_to_aggregate, replicas_to_aggregate=train_config.replicas_to_aggregate,
total_num_replicas=train_config.worker_replicas) total_num_replicas=worker_replicas)
sync_optimizer = training_optimizer sync_optimizer = training_optimizer
# Create ops required to initialize the model from a given checkpoint. # Create ops required to initialize the model from a given checkpoint.
......
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