Update davit.py

pull/1630/head
Fredo Guan 3 years ago
parent 96e0b06ee1
commit c1a5882c5a

@ -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):

Loading…
Cancel
Save