Update davit.py

3 years ago · b2beaa2364
parent 718508309f
commit b2beaa2364
1 changed files with 0 additions and 1 deletions
--- 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)