Add 21k maxvit tf weights

reinhud · May 11, 2023 · cf1884b · cf1884b
1 parent 6c2edf4
commit cf1884b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/timm/models/maxxvit.py b/timm/models/maxxvit.py
@@ -2013,23 +2013,26 @@ def _cfg(url='', **kwargs):
         input_size=(3, 512, 512), pool_size=(16, 16), crop_pct=1.0, crop_mode='squash'),
 
     'maxvit_base_tf_224.in21k': _cfg(
-        url=''),
+        hf_hub_id='timm/',
+        num_classes=21843),
     'maxvit_base_tf_384.in21k_ft_in1k': _cfg(
         hf_hub_id='timm/',
         input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'),
     'maxvit_base_tf_512.in21k_ft_in1k': _cfg(
         hf_hub_id='timm/',
         input_size=(3, 512, 512), pool_size=(16, 16), crop_pct=1.0, crop_mode='squash'),
     'maxvit_large_tf_224.in21k': _cfg(
-        url=''),
+        hf_hub_id='timm/',
+        num_classes=21843),
     'maxvit_large_tf_384.in21k_ft_in1k': _cfg(
         hf_hub_id='timm/',
         input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'),
     'maxvit_large_tf_512.in21k_ft_in1k': _cfg(
         hf_hub_id='timm/',
         input_size=(3, 512, 512), crop_pct=1.0, crop_mode='squash'),
     'maxvit_xlarge_tf_224.in21k': _cfg(
-        url=''),
+        hf_hub_id='timm/',
+        num_classes=21843),
     'maxvit_xlarge_tf_384.in21k_ft_in1k': _cfg(
         hf_hub_id='timm/',
         input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, crop_mode='squash'),