[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
# for free
to join this conversation on GitHub.
Already have an account?
# to comment
-
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