fix: Removed type annotation of forward method

Author: frgfm

torchvision/models/_utils.py

@@ -57,7 +57,7 @@ def __init__(self, model: nn.Module, return_layers: Dict[str, str]):
         super(IntermediateLayerGetter, self).__init__(layers)
         self.return_layers = orig_return_layers
 
-    def forward(self, x: torch.Tensor) -> Dict[str, torch.Tensor]:
+    def forward(self, x):
         out = OrderedDict()
         for name, module in self.items():
             x = module(x)