From 2b5f39114206b27040b20e10bf9d0d2d988da937 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Fri, 16 Dec 2022 16:52:22 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index e44519df..0ccd2ae0 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -473,13 +473,7 @@ class DaViT(nn.Module): self.grad_checkpointing = False self.feature_info = [] - self.patch_embed = PatchEmbed( - patch_size=patch_size, - in_chans=in_chans, - embed_dim=embed_dims[0], - overlapped=overlapped_patch - ) - + self.patch_embed = None stages = [] for stage_id in range(self.num_stages): @@ -503,6 +497,7 @@ class DaViT(nn.Module): ) if stage_id == 0: + self.patch_embed = stage.patch_embed stage.patch_embed = nn.Identity() stages.append(stage)