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