Skip to content

GitLab

Commit ddbf3833 authored by Yuhao Zhang's avatar Yuhao Zhang Committed by Toby Boyd
Browse files

specify the 'num_gpus' argument (#7157)

parent 3768f4cf
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/deep_speech/deep_speech.py
View file @ ddbf3833
...@@ -223,7 +223,7 @@ def run_deep_speech(_): ...@@ -223,7 +223,7 @@ def run_deep_speech(_):
# Use distribution strategy for multi-gpu training # Use distribution strategy for multi-gpu training
num_gpus = flags_core.get_num_gpus(flags_obj) num_gpus = flags_core.get_num_gpus(flags_obj)
distribution_strategy = distribution_utils.get_distribution_strategy(num_gpus) distribution_strategy = distribution_utils.get_distribution_strategy(num_gpus=num_gpus)
run_config = tf.estimator.RunConfig( run_config = tf.estimator.RunConfig(
train_distribute=distribution_strategy) train_distribute=distribution_strategy)
......
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