@ -37,8 +37,8 @@ __all__ = ['DaViT']
class SequentialWithSize(nn.Sequential):
def forward(self, x: Tensor, size: Tuple[int, int]):
for module in self:
output = module(inputs, size)
inputs : Tensor = output[0]
output = module(x, size)
x : Tensor = output[0]
size : Tuple[int, int] = output[1]
return x, size