Update davit.py

3 years ago · 3cff8d0ff4
parent 1bff091c78
commit 3cff8d0ff4
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -411,7 +411,7 @@ class DaViTStage(nn.Module):
        self.blocks = nn.Sequential(*stage_blocks)
        
    def forward(self, x : Tensor):
-        x = self.patch_embed(x)
+        #x = self.patch_embed(x)
        if self.grad_checkpointing and not torch.jit.is_scripting():
            x = checkpoint_seq(self.blocks, x)
        else: