Update davit.py

timm/models/davit.py

@@ -37,7 +37,7 @@ __all__ = ['DaViT']
 
 class MySequential(nn.Sequential):
     def forward(self, inputs : Tuple[Tensor, Tuple[int, int]]):
-        for module in self._modules.values():
+        for module in self:
             #if type(inputs) == tuple:
             inputs = module(*inputs)
             #else: