Fix RDNet forward call

2025-01-22 00:41:04 +08:00 · 2025-01-22 00:41:04 +08:00 · 61bf9e1af1
parent c0b12fcb16
commit 61bf9e1af1
1 changed files with 1 additions and 1 deletions
--- a/timm/models/rdnet.py
+++ b/timm/models/rdnet.py
@ -362,7 +362,7 @@ class RDNet(nn.Module):

    def forward(self, x):
        x = self.forward_features(x)
-        x = self.head(x)
+        x = self.forward_head(x)
        return x

    @torch.jit.ignore