Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "81a34cbe2d40effb31b3abf490a7cb0ed6dd5b92"
Commit 437e6318 authored by Xianzhi Du's avatar Xianzhi Du Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 389310818
parent 2af488b2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/beta/modeling/layers/nn_layers.py
View file @ 437e6318
...@@ -165,7 +165,8 @@ class SqueezeExcitation(tf.keras.layers.Layer): ...@@ -165,7 +165,8 @@ class SqueezeExcitation(tf.keras.layers.Layer):
def build(self, input_shape): def build(self, input_shape):
num_reduced_filters = make_divisible( num_reduced_filters = make_divisible(
self._in_filters * self._se_ratio, divisor=self._divisible_by) max(1, int(self._in_filters * self._se_ratio)),
divisor=self._divisible_by)
self._se_reduce = tf.keras.layers.Conv2D( self._se_reduce = tf.keras.layers.Conv2D(
filters=num_reduced_filters, filters=num_reduced_filters,
......
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