Skip to content

GitLab

Unverified Commit ed6c805a authored by Shining Sun's avatar Shining Sun Committed by GitHub
Browse files

Flag (#6121) 

* Fix the turn_off_ds flag problem

* add param names to all args
parent 23721cf2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
official/resnet/keras/keras_cifar_main.py
View file @ ed6c805a
...@@ -136,7 +136,8 @@ def run(flags_obj): ...@@ -136,7 +136,8 @@ def run(flags_obj):
parse_record_fn=parse_record_keras) parse_record_fn=parse_record_keras)
strategy = distribution_utils.get_distribution_strategy( strategy = distribution_utils.get_distribution_strategy(
flags_obj.num_gpus, flags_obj.turn_off_distribution_strategy) num_gpus=flags_obj.num_gpus,
turn_off_distribution_strategy=flags_obj.turn_off_distribution_strategy)
strategy_scope = keras_common.get_strategy_scope(strategy) strategy_scope = keras_common.get_strategy_scope(strategy)
......
official/resnet/keras/keras_imagenet_main.py
View file @ ed6c805a
...@@ -125,7 +125,8 @@ def run(flags_obj): ...@@ -125,7 +125,8 @@ def run(flags_obj):
parse_record_fn=parse_record_keras) parse_record_fn=parse_record_keras)
strategy = distribution_utils.get_distribution_strategy( strategy = distribution_utils.get_distribution_strategy(
flags_obj.num_gpus, flags_obj.turn_off_distribution_strategy) num_gpus=flags_obj.num_gpus,
turn_off_distribution_strategy=flags_obj.turn_off_distribution_strategy)
strategy_scope = keras_common.get_strategy_scope(strategy) strategy_scope = keras_common.get_strategy_scope(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