From fb3a9528f40e65531a35f51d8087532bc3b18b85 Mon Sep 17 00:00:00 2001
From: Nrupatunga <nrupatunga.tunga@gmail.com>
Date: Thu, 7 Apr 2022 17:22:44 +0530
Subject: [PATCH] Fix Tf export for BottleneckCSP (#7330)

---
 models/tf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/tf.py b/models/tf.py
index a15569e3b465..04b1cd378f18 100644
--- a/models/tf.py
+++ b/models/tf.py
@@ -152,7 +152,7 @@ def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5, w=None):
         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):