From 38ea27001fac5cc7120026397c7dbe7a3fc26371 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 06:27:51 -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 1de48da8..a4fce297 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -421,7 +421,7 @@ class DaViT(nn.Module): for stage_id, stage_param in enumerate(self.architecture): layer_offset_id = len(list(itertools.chain(*self.architecture[:stage_id]))) - stage = nn.ModuleList([ + stage = MySequential(*[ MySequential(*[ ChannelBlock( dim=self.embed_dims[item],