Update davit.py

3 years ago · c1a5882c5a
parent 96e0b06ee1
commit c1a5882c5a
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -510,7 +510,7 @@ class DaViT(nn.Module):
-        for patch_layer, blocks in itertools.izip(self.patch_embeds, self.main_blocks):
+        for patch_layer, blocks in zip(self.patch_embeds, self.main_blocks):
            features[-1], sizes[-1] = patch_layer(features[-1], sizes[-1])
            for layer in enumerate(blocks):