Update davit.py

3 years ago · 1bff091c78
parent c6973a3ebe
commit 1bff091c78
1 changed files with 1 additions and 1 deletions
--- 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}')]