diff --git a/tests/test_models.py b/tests/test_models.py index 030fb255..15e6cc35 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -60,7 +60,7 @@ NON_STD_FILTERS = [ 'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*', 'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*', 'poolformer_*', 'volo_*', 'sequencer2d_*', 'mvitv2*', 'gcvit*', 'efficientformer*', - 'eva_*', 'flexivit*', 'eva02*', 'samvit_*', 'efficientvit_m*', 'tiny_vit_*', 'hiera_*', 'vitamin*' + 'eva_*', 'flexivit*', 'eva02*', 'samvit_*', 'efficientvit_m*', 'tiny_vit_*', 'hiera_*', 'vitamin*', 'test_vit*', ] NUM_NON_STD = len(NON_STD_FILTERS) diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index fe0460dc..f2cd8f34 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -2285,7 +2285,7 @@ default_cfgs = generate_default_cfgs({ hf_hub_filename='open_clip_pytorch_model.bin', num_classes=1024, mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, fixed_input_size=True, input_size=(3, 224, 224), pool_size=(7, 7), - classifier = 'head.proj', + classifier='head.proj', ), 'resnet101_clip.openai': _cfgr( hf_hub_id='timm/', @@ -2299,21 +2299,21 @@ default_cfgs = generate_default_cfgs({ hf_hub_filename='open_clip_pytorch_model.bin', num_classes=640, mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, fixed_input_size=True, input_size=(3, 288, 288), pool_size=(9, 9), - classifier = 'head.proj', + classifier='head.proj', ), 'resnet50x16_clip.openai': _cfgr( hf_hub_id='timm/', hf_hub_filename='open_clip_pytorch_model.bin', num_classes=768, mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, fixed_input_size=True, input_size=(3, 384, 384), pool_size=(12, 12), - classifier = 'head.proj', + classifier='head.proj', ), 'resnet50x64_clip.openai': _cfgr( hf_hub_id='timm/', hf_hub_filename='open_clip_pytorch_model.bin', num_classes=1024, mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, fixed_input_size=True, input_size=(3, 448, 448), pool_size=(14, 14), - classifier = 'head.proj', + classifier='head.proj', ), # avg-pool w/ optional standard classifier head variants @@ -2354,6 +2354,7 @@ default_cfgs = generate_default_cfgs({ 'test_byobnet.untrained': _cfgr( # hf_hub_id='timm/', + first_conv='stem.conv', input_size=(3, 160, 160), crop_pct=0.875, pool_size=(5, 5), ), })