Skip to content

Commit b3869b6

Browse files
committed
Update comments for mnv5 pretrained configs
1 parent 6c5e3ce commit b3869b6

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

timm/models/mobilenetv5.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -815,19 +815,20 @@ def _cfg(url: str = '', **kwargs):
815815

816816

817817
default_cfgs = generate_default_cfgs({
818-
# encoder-only configs
818+
# Encoder-only config for Gemma 3n Transformers integration
819819
'mobilenetv5_300m_enc': _cfg(
820-
#hf_hub_id='timm/',
821820
mean=(0., 0., 0.), std=(1., 1., 1.),
822821
input_size=(3, 768, 768),
823822
num_classes=0),
824823

825-
# WIP classification configs for testing
824+
# Gemma 3n encoder weights for timm use / fine-tune
826825
'mobilenetv5_300m.gemma3n': _cfg(
827826
hf_hub_id='timm/',
828827
mean=(0., 0., 0.), std=(1., 1., 1.),
829828
input_size=(3, 768, 768),
830829
num_classes=0),
830+
831+
# WIP classification configs for testing
831832
'mobilenetv5_base.untrained': _cfg(
832833
# hf_hub_id='timm/',
833834
num_classes=1000)

0 commit comments

Comments
 (0)