Update davit.py

3 years ago · e5b75570c8
parent dda97432c7
commit e5b75570c8
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -477,7 +477,7 @@ class DaViT(nn.Module):
        features = [x]
        sizes = [size]
-        for patch_layer, stage in enumerate(zip(self.patch_embeds, self.stages)):
+        for patch_layer, stage in zip(self.patch_embeds, self.stages):
            features[-1], sizes[-1] = patch_layer(features[-1], sizes[-1])
            for _, block in enumerate(stage):
                for _, layer in enumerate(block):