Incorrectly named head_hidden_size

rwightman · rwightman · commit b0cfd9de6d7e · 2024-09-13T18:21:05.000-07:00
diff --git a/timm/models/mambaout.py b/timm/models/mambaout.py
@@ -384,7 +384,7 @@ def __init__(
                 drop_rate=drop_rate,
             )
         self.num_features = prev_dim
-        self.hidden_size = self.head.num_features
+        self.head_hidden_size = self.head.num_features
 
         self.apply(self._init_weights)
 

Original file line number	Diff line number	Diff line change
`@@ -384,7 +384,7 @@ def __init__(`
`384`	`384`	`drop_rate=drop_rate,`
`385`	`385`	`)`
`386`	`386`	`self.num_features = prev_dim`
`387`		`- self.hidden_size = self.head.num_features`
	`387`	`+ self.head_hidden_size = self.head.num_features`
`388`	`388`
`389`	`389`	`self.apply(self._init_weights)`
`390`	`390`