Nrupatunga
commited on
Commit
•
b7faeda
1
Parent(s):
0ca85ed
Fix Tf export for BottleneckCSP (#7330)
Browse files- 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.
|
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):
|