From 089d3f24e779b97e8b6e30d1f15b7ce68e99c370 Mon Sep 17 00:00:00 2001
From: Ahmed Mesih <74631008+AhmedMesih@users.noreply.github.com>
Date: Fri, 9 Jul 2021 00:42:44 +0300
Subject: [PATCH] update

---
 utils/layers.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils/layers.py b/utils/layers.py
index 1d46a18..e788711 100644
--- a/utils/layers.py
+++ b/utils/layers.py
@@ -11,7 +11,8 @@
 except:
     class Mish(nn.Module):  # https://github.com/digantamisra98/Mish
         def forward(self, x):
-            return x * F.softplus(x).tanh()
+            # return x * F.softplus(x).tanh()
+            return F.mish(x)    
 
 
 class Reorg(nn.Module):