From 143a239e93ad18d00d6b2d0bb4fe5e0434969696 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Tue, 6 Dec 2022 21:09:05 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 72c11679..e9957f19 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -70,6 +70,7 @@ class ConvPosEnc(nn.Module):
                               to_2tuple(k // 2),
                               groups=dim)
         self.normtype = normtype
+        self.norm = nn.Identity()
         if self.normtype == 'batch':
             self.norm = nn.BatchNorm2d(dim)
         elif self.normtype == 'layer':