Skip to content

GitLab

Unverified Commit 68650c42 authored by rxsang's avatar rxsang Committed by GitHub
Browse files

Fix non dist strat case. (#6867)

parent abe9e96a
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 6 deletions
official/resnet/keras/keras_imagenet_main.py
View file @ 68650c42
...@@ -129,12 +129,13 @@ def run(flags_obj): ...@@ -129,12 +129,13 @@ def run(flags_obj):
all_reduce_alg=flags_obj.all_reduce_alg, all_reduce_alg=flags_obj.all_reduce_alg,
num_packs=flags_obj.num_packs) num_packs=flags_obj.num_packs)
# flags_obj.enable_get_next_as_optional controls whether enabling if strategy:
# get_next_as_optional behavior in DistributedIterator. If true, last partial # flags_obj.enable_get_next_as_optional controls whether enabling
# batch can be supported. # get_next_as_optional behavior in DistributedIterator. If true, last
strategy.extended.experimental_enable_get_next_as_optional = ( # partial batch can be supported.
flags_obj.enable_get_next_as_optional strategy.extended.experimental_enable_get_next_as_optional = (
) flags_obj.enable_get_next_as_optional
)
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