[BUG] Wrong model.num_features for mobilenetv4 #2291
Unanswered
MichaelMonashev
asked this question in
Q&A
Replies: 1 comment
-
@MichaelMonashev it is not wrong https://huggingface.co/timm/mobilenetv4_conv_small.e2400_r224_in1k/discussions/3
head_hidden_size != num_features for models like mobilenetv3/v4 or some other models that have mlps or other conv/linear layers in the head AFTER pooling |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Describe the bug
model.num_features
return 960, but correct number is 1280.To Reproduce
Expected behavior
Beta Was this translation helpful? Give feedback.
All reactions