Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e93fdccac0ded9117331cf93f378b366aa6e7577"
Commit a7338771 authored by Haoyu Zhang's avatar Haoyu Zhang Committed by Toby Boyd
Browse files

Revert "Specify NCCL as the all reduce algorithm (#6662)" (#6671) 

Reason: test failures because contrib is not available in V2

This reverts commit 325dd761.
parent ff5cef9a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/resnet/keras/keras_imagenet_main.py
View file @ a7338771
...@@ -124,8 +124,7 @@ def run(flags_obj): ...@@ -124,8 +124,7 @@ def run(flags_obj):
strategy = distribution_utils.get_distribution_strategy( strategy = distribution_utils.get_distribution_strategy(
distribution_strategy=flags_obj.distribution_strategy, distribution_strategy=flags_obj.distribution_strategy,
num_gpus=flags_obj.num_gpus, num_gpus=flags_obj.num_gpus,
num_workers=distribution_utils.configure_cluster(), num_workers=distribution_utils.configure_cluster())
all_reduce_alg='nccl')
strategy_scope = distribution_utils.get_strategy_scope(strategy) strategy_scope = distribution_utils.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