From a813bbd1df59668785d6ae7968420829217139a0 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Wed, 7 Dec 2022 00:53:53 -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 270f687c..4a3c03e7 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -514,7 +514,7 @@ class DaViT(nn.Module): features[-1], sizes[-1] = patch_layer(features[-1], sizes[-1]) for block in enumerate(stage): - for layer in enumerate(block): + for _, layer in enumerate(block): if self.grad_checkpointing and not torch.jit.is_scripting(): features[-1], sizes[-1] = checkpoint.checkpoint(layer, (features[-1], sizes[-1])) else: