diff --git a/timm/models/davit.py b/timm/models/davit.py index c815cb90..061a68bf 100644 --- 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: