Update davit.py

3 years ago · f5da145a76
parent 3a2dc4fbd8
commit f5da145a76
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -419,7 +419,7 @@ class DaViT(nn.Module):
            layer_offset_id = len(list(itertools.chain(*self.architecture[:stage_id])))

            stage = MySequential(*[
-                Sequential(*[
+                MySequential(*[
                    ChannelBlock(
                        dim=self.embed_dims[item],
                        num_heads=self.num_heads[item],