From 4303034189027f5df5da1cd50e3d558bd8d4444f Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 06:02:30 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 582e4010..09b44e37 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -38,6 +38,8 @@ class MySequential(nn.Sequential):
 
     def forward(self, inputs : Tensor, size : Tuple[int, int]):
         for module in self:
+            inputs : Tensor
+            size : Tuple[int, int]
             (inputs, size) = module(inputs, size)
         return inputs