Nrupatunga commited on
Commit
b7faeda
1 Parent(s): 0ca85ed

Fix Tf export for BottleneckCSP (#7330)

Browse files
Files changed (1) hide show
  1. models/tf.py +1 -1
models/tf.py CHANGED
@@ -152,7 +152,7 @@ class TFBottleneckCSP(keras.layers.Layer):
152
  self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3)
153
  self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4)
154
  self.bn = TFBN(w.bn)
155
- self.act = lambda x: keras.activations.relu(x, alpha=0.1)
156
  self.m = keras.Sequential([TFBottleneck(c_, c_, shortcut, g, e=1.0, w=w.m[j]) for j in range(n)])
157
 
158
  def call(self, inputs):
 
152
  self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3)
153
  self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4)
154
  self.bn = TFBN(w.bn)
155
+ self.act = lambda x: keras.activations.swish(x)
156
  self.m = keras.Sequential([TFBottleneck(c_, c_, shortcut, g, e=1.0, w=w.m[j]) for j in range(n)])
157
 
158
  def call(self, inputs):