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

correct summing total blocks

This commit is contained in:
Yixing Fu 2020-06-05 18:55:32 -04:00
parent d54578dba2
commit d8bb6569bc
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.