Remove debug / staging code

Author: rwightman

timm/models/eva.py

@@ -717,11 +717,6 @@ def checkpoint_filter_fn(
             # fixed embedding no need to load buffer from checkpoint
             continue
 
-        # FIXME here while importing new weights, to remove
-        # if k == 'cls_token':
-        #     print('DEBUG: cls token -> reg')
-        #     k = 'reg_token'
-
         if 'patch_embed.proj.weight' in k:
             _, _, H, W = model.patch_embed.proj.weight.shape
             if v.shape[-1] != W or v.shape[-2] != H:
@@ -952,24 +947,20 @@ def _cfg(url='', **kwargs):
 
     'vit_medium_patch16_rope_reg1_gap_256.sbb_in1k': _cfg(
         hf_hub_id='timm/',
-        #file='vit_medium_gap1_rope-in1k-20230920-5.pth',
         input_size=(3, 256, 256), crop_pct=0.95,
         mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)
     ),
     'vit_mediumd_patch16_rope_reg1_gap_256.sbb_in1k': _cfg(
         hf_hub_id='timm/',
-        #file='vit_mediumd_gap1_rope-in1k-20230926-5.pth',
         input_size=(3, 256, 256), crop_pct=0.95,
         mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)
     ),
     'vit_betwixt_patch16_rope_reg4_gap_256.sbb_in1k': _cfg(
         hf_hub_id='timm/',
-        #file='vit_betwixt_gap4_rope-in1k-20231005-5.pth',
         input_size=(3, 256, 256), crop_pct=0.95,
     ),
     'vit_base_patch16_rope_reg1_gap_256.sbb_in1k': _cfg(
         hf_hub_id='timm/',
-        #file='vit_base_gap1_rope-in1k-20230930-5.pth',
         input_size=(3, 256, 256), crop_pct=0.95,
         mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)
     ),

timm/models/vision_transformer.py

@@ -1791,47 +1791,36 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
         mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, num_classes=512),
 
     'vit_wee_patch16_reg1_gap_256.sbb_in1k': _cfg(
-        #file='',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_pwee_patch16_reg1_gap_256.sbb_in1k': _cfg(
-        #file='./vit_pwee-in1k-8.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_little_patch16_reg4_gap_256.sbb_in1k': _cfg(
-        #file='vit_little_patch16-in1k-8a.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_medium_patch16_reg1_gap_256.sbb_in1k': _cfg(
-        #file='vit_medium_gap1-in1k-20231118-8.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_medium_patch16_reg4_gap_256.sbb_in1k': _cfg(
-        #file='vit_medium_gap4-in1k-20231115-8.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_mediumd_patch16_reg4_gap_256.sbb_in12k_ft_in1k': _cfg(
-        #file='vit_mp_patch16_reg4-in1k-5a.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_mediumd_patch16_reg4_gap_256.sbb_in12k': _cfg(
-        #file='vit_mp_patch16_reg4-in12k-8.pth',
         hf_hub_id='timm/',
         num_classes=11821,
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_betwixt_patch16_reg1_gap_256.sbb_in1k': _cfg(
-        #file='vit_betwixt_gap1-in1k-20231121-8.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_betwixt_patch16_reg4_gap_256.sbb_in12k_ft_in1k': _cfg(
-        #file='vit_betwixt_patch16_reg4-ft-in1k-8b.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_betwixt_patch16_reg4_gap_256.sbb_in1k': _cfg(
-        #file='vit_betwixt_gap4-in1k-20231106-8.pth',
         hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_betwixt_patch16_reg4_gap_256.sbb_in12k': _cfg(
-        #file='vit_betwixt_gap4-in12k-8.pth',
         hf_hub_id='timm/',
         num_classes=11821,
         input_size=(3, 256, 256), crop_pct=0.95),