Skip to content

Commit cf1884b

Browse files
committed
Add 21k maxvit tf weights
1 parent 6c2edf4 commit cf1884b

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

timm/models/maxxvit.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2013,23 +2013,26 @@ def _cfg(url='', **kwargs):
20132013
input_size=(3, 512, 512), pool_size=(16, 16), crop_pct=1.0, crop_mode='squash'),
20142014

20152015
'maxvit_base_tf_224.in21k': _cfg(
2016-
url=''),
2016+
hf_hub_id='timm/',
2017+
num_classes=21843),
20172018
'maxvit_base_tf_384.in21k_ft_in1k': _cfg(
20182019
hf_hub_id='timm/',
20192020
input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'),
20202021
'maxvit_base_tf_512.in21k_ft_in1k': _cfg(
20212022
hf_hub_id='timm/',
20222023
input_size=(3, 512, 512), pool_size=(16, 16), crop_pct=1.0, crop_mode='squash'),
20232024
'maxvit_large_tf_224.in21k': _cfg(
2024-
url=''),
2025+
hf_hub_id='timm/',
2026+
num_classes=21843),
20252027
'maxvit_large_tf_384.in21k_ft_in1k': _cfg(
20262028
hf_hub_id='timm/',
20272029
input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'),
20282030
'maxvit_large_tf_512.in21k_ft_in1k': _cfg(
20292031
hf_hub_id='timm/',
20302032
input_size=(3, 512, 512), crop_pct=1.0, crop_mode='squash'),
20312033
'maxvit_xlarge_tf_224.in21k': _cfg(
2032-
url=''),
2034+
hf_hub_id='timm/',
2035+
num_classes=21843),
20332036
'maxvit_xlarge_tf_384.in21k_ft_in1k': _cfg(
20342037
hf_hub_id='timm/',
20352038
input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'),

0 commit comments

Comments
 (0)