Internal change

PiperOrigin-RevId: 377424053

Internal change
PiperOrigin-RevId: 377424053
1ba20e04 · A. Unique TensorFlower · 8d0c7942 · 1ba20e04
Commit 1ba20e04 authored Jun 03, 2021 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

official/vision/beta/modeling/layers/nn_layers.py official/vision/beta/modeling/layers/nn_layers.py +1 -2

No files found.
--- a/official/vision/beta/modeling/layers/nn_layers.py
+++ b/official/vision/beta/modeling/layers/nn_layers.py
@@ -132,8 +132,7 @@ class SqueezeExcitation(tf.keras.layers.Layer):

  def build(self, input_shape):
    num_reduced_filters = make_divisible(
-        max(1, int(self._in_filters * self._se_ratio)),
-        divisor=self._divisible_by)
+        self._in_filters * self._se_ratio, divisor=self._divisible_by)

    self._se_reduce = tf.keras.layers.Conv2D(
        filters=num_reduced_filters,