diff --git a/timm/models/davit.py b/timm/models/davit.py index fe3cbad4..ae5694d4 100644 --- a/timm/models/davit.py +++ b/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: