From 625bb86a644745412dc7da6eeb8bbf8185c4d161 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Wed, 7 Dec 2022 01:01:22 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index bd9e0e09..440c912d 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -512,10 +512,9 @@ class DaViT(nn.Module): for patch_layer, stage in zip(self.patch_embeds, self.main_blocks): features[-1], sizes[-1] = patch_layer(features[-1], sizes[-1]) - print(stage) - for block in enumerate(stage): + for _, block in enumerate(stage): print(block) - for layer in enumerate(block): + for _, layer in enumerate(block): print(layer) if self.grad_checkpointing and not torch.jit.is_scripting(): features[-1], sizes[-1] = checkpoint.checkpoint(layer, (features[-1], sizes[-1]))