From c1a5882c5a2b23ad5e99a3b800345726e2ebe006 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Wed, 7 Dec 2022 00:39:37 -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 41af0ebe..464b976f 100644 --- 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):