From e5b75570c82d90ece7685dccd03f05aefcb87153 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 04:45:11 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 0259ec9f..33fd403e 100644
--- 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):