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

Merge pull request #43880 from andreABbauer:patch-1

Browse Source 
PiperOrigin-RevId: 343064483
Change-Id: I4f2cf586ba8ff2fed598bec6c2763762a838579a
This commit is contained in:
TensorFlower Gardener 2020-11-18 06:19:29 -08:00
commit 402d28705e
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -479,7 +479,11 @@ def block(inputs,
if 0 < se_ratio <= 1:
filters_se = max(1, int(filters_in * se_ratio))
se = layers.GlobalAveragePooling2D(name=name + 'se_squeeze')(x)
se = layers.Reshape((1, 1, filters), name=name + 'se_reshape')(se)
if bn_axis == 1:
se_shape = (filters, 1, 1)
else:
se_shape = (1, 1, filters)
se = layers.Reshape(se_shape, name=name + 'se_reshape')(se)
se = layers.Conv2D(
filters_se,
1,