diff --git a/timm/models/davit.py b/timm/models/davit.py index c2f3fb2b..c815cb90 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -502,7 +502,7 @@ class DaViT(nn.Module): cpe_act = cpe_act ) - stages.append(patch_embed + stages.append(patch_embed) stages.append(stage) self.feature_info += [dict(num_chs=self.embed_dims[stage_id], reduction=2, module=f'stages.{stage_id}')]