experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #40210 from yixingfu:master 

PiperOrigin-RevId: 317329321
Change-Id: If87b7e597359984da3ace62c6f48281602b8653d
This commit is contained in:
TensorFlower Gardener 2020-06-19 10:18:08 -07:00
parent af5b7537d1 5869f65762
commit 7bd8698f34
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/applications/efficientnet.py
View File

@ -334,7 +334,7 @@ def EfficientNet(
blocks_args = copy.deepcopy(blocks_args) blocks_args = copy.deepcopy(blocks_args)
b = 0 b = 0
blocks = float(sum(args['repeats'] for args in blocks_args)) blocks = float(sum(round_repeats(args['repeats']) for args in blocks_args))
for (i, args) in enumerate(blocks_args): for (i, args) in enumerate(blocks_args):
assert args['repeats'] > 0 assert args['repeats'] > 0
# Update block input and output filters based on depth multiplier. # Update block input and output filters based on depth multiplier.