From 8bb6b3687bb8c3427458b6a8e617021848ffa917 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 11:01:43 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 3aab3f6e..a2c39a49 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -108,8 +108,6 @@ class PatchEmbed(nn.Module):
 
     def forward(self, x : Tensor, size: Tuple[int, int]):
         H, W = size
-        #dim = x.dim()
-        #if dim == 3:
         in_shape = x.shape
         
         if self.norm_after == False:
@@ -128,7 +126,6 @@ class PatchEmbed(nn.Module):
         x = self.proj(x)
         newsize = (x.size(2), x.size(3))
         x = x.flatten(2).transpose(1, 2)
-        #if dim == 4:
         if self.norm_after == True:
             x = self.norm(x)
         return x, newsize