From f47f6fce31dd7cff0476d61d68e9f16edf16cdd5 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 05:23:38 -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 97d644fc..b8a9bb66 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -411,8 +411,8 @@ 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([
-                nn.ModuleList([
+            stage = nn.Sequential([
+                nn.Sequential([
                     ChannelBlock(
                         dim=self.embed_dims[item],
                         num_heads=self.num_heads[item],