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

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

diff --git a/timm/models/davit.py b/timm/models/davit.py
index c006fbce..48b387c4 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -55,7 +55,6 @@ class ConvPosEnc(nn.Module):
     def forward(self, x : Tensor, size: Tuple[int, int]):
         B, N, C = x.shape
         H, W = size
-        assert N == H * W
 
         feat = x.transpose(1, 2).view(B, C, H, W)
         feat = self.proj(feat)