diff --git a/timm/models/hrnet.py b/timm/models/hrnet.py index aee2f99fb7..75b157d67d 100644 --- a/timm/models/hrnet.py +++ b/timm/models/hrnet.py @@ -834,7 +834,7 @@ def __init__( def forward_features(self, x): assert False, 'Not supported' - def forward(self, x) -> List[torch.tensor]: + def forward(self, x) -> List[torch.Tensor]: out = [] x = self.conv1(x) x = self.bn1(x)