From 222dca8f7b177582d6f0c0a80c5a674bf0df483a Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Tue, 6 Dec 2022 22:14:15 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index af61eda0..f4499855 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -507,8 +507,8 @@ class DaViT(nn.Module): branches = [0] for block_index, block_param in enumerate(self.architecture): - branch_ids = sorted(set(torch.Tensor(block_param))) - #branch_ids = [block_index] + #branch_ids = sorted(set(block_param)) + branch_ids = [block_index] for branch_id in branch_ids: if branch_id not in branches: x, size = self.patch_embeds[branch_id](features[-1], sizes[-1])