From f7a8fb9f97d1ef7b17d25883a3207649cf29f4c9 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 00:26:27 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index 22a12470..9fb8fdea 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -494,9 +494,7 @@ 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]) for _, block in enumerate(stage): - print(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]) else: