Fix Tf export for BottleneckCSP (#7330)

This commit is contained in:
Nrupatunga authored and GitHub committed 2022-04-07 13:52:44 +02:00
1 parent 0ca85ed65f
commit b7faeda0f2
1 file changed
+1 -1
+1 -1
View File
@@ -152,7 +152,7 @@ class TFBottleneckCSP(keras.layers.Layer):
self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3)
self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4)
self.bn = TFBN(w.bn)
self.act = lambda x: keras.activations.relu(x, alpha=0.1)
self.act = lambda x: keras.activations.swish(x)
self.m = keras.Sequential([TFBottleneck(c_, c_, shortcut, g, e=1.0, w=w.m[j]) for j in range(n)])
def call(self, inputs):