Initial attempt to populate HF-related metadata

brainscore_vision/models/AdvProp_efficientnet_b2/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 351
         trainable_layers: 116
         model_size_mb: 36.44
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/AdvProp_efficientnet_b2
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/smp-hub/efficientnet-b2.advprop
         extra_notes: null
         runnable: true

brainscore_vision/models/AdvProp_efficientnet_b4/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 486
         trainable_layers: 161
         model_size_mb: 77.37
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/AdvProp_efficientnet_b4
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/smp-hub/efficientnet-b4.advprop
         extra_notes: null
         runnable: true

brainscore_vision/models/AdvProp_efficientnet_b6/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 678
         trainable_layers: 225
         model_size_mb: 172.16
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/AdvProp_efficientnet_b6
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/smp-hub/efficientnet-b6.advprop
         extra_notes: null
         runnable: true

brainscore_vision/models/AdvProp_efficientnet_b7/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 825
         trainable_layers: 274
         model_size_mb: 265.39
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/AdvProp_efficientnet_b7
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/smp-hub/efficientnet-b7.advprop
         extra_notes: null
         runnable: true

brainscore_vision/models/AdvProp_efficientnet_b8/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 915
         trainable_layers: 304
         model_size_mb: 349.65
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/AdvProp_efficientnet_b8
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/smp-hub/timm-efficientnet-b8.advprop
         extra_notes: null
         runnable: true

brainscore_vision/models/Res2Net50_26w_4s/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 251
         trainable_layers: 86
         model_size_mb: 102.8
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/Res2Net50_26w_4s
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/res2net50_26w_4s.in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/alexnet/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 24
         trainable_layers: 8
         model_size_mb: 244.4
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/alexnet
-        huggingface_link: https://huggingface.co/alexnet
+        huggingface_link: null
         extra_notes: null
         runnable: true

brainscore_vision/models/alexnet_training_seed_03/metadata.yaml

@@ -8,7 +8,7 @@ models:
         trainable_layers: 8
         model_size_mb: 201.22
         training_dataset: null
-        task_specialization: null
+        task_specialization: text-generation
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/alexnet_training_seed_03
         huggingface_link: null
         extra_notes: null

brainscore_vision/models/antialias_resnet152/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 441
         trainable_layers: 156
         model_size_mb: 240.77
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/antialias_resnet152
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet152.a1h_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/artResNet18_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 68
         trainable_layers: 21
         model_size_mb: 46.76
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/artResNet18_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet18.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/barlow_twins_custom/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 68
         trainable_layers: 21
         model_size_mb: 46.76
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/barlow_twins_custom
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet18.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/blur_timm_models/metadata.yaml

@@ -7,10 +7,10 @@ models:
         total_layers: 465
         trainable_layers: 113
         model_size_mb: 1400.79
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; imagenet-22k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/convnext_large.fb_in22k_ft_in1k
         extra_notes: null
         runnable: true
     blur25_convnext_xxlarge:clip_laion2b_soup_ft_in1k:
@@ -21,10 +21,10 @@ models:
         total_layers: 513
         trainable_layers: 125
         model_size_mb: 3385.88
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; laion-2b
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/convnext_xxlarge.clip_laion2b_soup_ft_in1k
         extra_notes: null
         runnable: true
     blur25_vit_huge_patch14_clip_224:laion2b_ft_in12k_in1k:
@@ -35,10 +35,10 @@ models:
         total_layers: 684
         trainable_layers: 132
         model_size_mb: 2528.19
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; laion-2b; imagenet-12k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/vit_huge_patch14_clip_224.laion2b_ft_in12k_in1k
         extra_notes: null
         runnable: true
     blur25_vit_large_patch14_clip_224:laion2b_ft_in12k_in1k:
@@ -49,10 +49,10 @@ models:
         total_layers: 516
         trainable_layers: 100
         model_size_mb: 1216.82
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; laion-2b; imagenet-12k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/vit_large_patch14_clip_224.laion2b_ft_in12k_in1k
         extra_notes: null
         runnable: true
     blur25_vit_large_patch14_clip_224:openai_ft_in12k_in1k:
@@ -63,10 +63,10 @@ models:
         total_layers: 516
         trainable_layers: 100
         model_size_mb: 1216.82
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; wit-400m; imagenet-12k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/vit_large_patch14_clip_224.openai_ft_in12k_in1k
         extra_notes: null
         runnable: true
     blur25_vit_large_patch14_clip_336:openai_ft_in12k_in1k:
@@ -77,10 +77,10 @@ models:
         total_layers: 516
         trainable_layers: 100
         model_size_mb: 1218.13
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; wit-400m; imagenet-12k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/vit_large_patch14_clip_336.openai_ft_in12k_in1k
         extra_notes: null
         runnable: true
     blur25_convnext_large_mlp:clip_laion2b_augreg_ft_in1k_384:
@@ -91,10 +91,10 @@ models:
         total_layers: 467
         trainable_layers: 114
         model_size_mb: 800.51
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; laion-2b
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384
         extra_notes: null
         runnable: true
     blur25_convnext_tiny:in12k_ft_in1k:
@@ -105,9 +105,9 @@ models:
         total_layers: 249
         trainable_layers: 59
         model_size_mb: 114.36
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k; imagenet-12k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/blur_timm_models
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/convnext_tiny.in12k_ft_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/cifar_resnet18_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 68
         trainable_layers: 21
         model_size_mb: 46.73
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/cifar_resnet18_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet18.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/densenet_201/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 919
         trainable_layers: 201
         model_size_mb: 80.06
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/densenet_201
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/densenet201.tv_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_Vanilla/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_Vanilla
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_Vanilla_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_Vanilla_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_Vanilla_1_full/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_Vanilla_1_full
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_Vanilla_2/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_Vanilla_2
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_augself_linear_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_augself_linear_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_augself_mlp_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_augself_mlp_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_lmda_0001_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_lmda_0001_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_lmda_001_1/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_lmda_001_1
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true

brainscore_vision/models/eBarlow_lmda_001_2/metadata.yaml

@@ -7,9 +7,9 @@ models:
         total_layers: 151
         trainable_layers: 54
         model_size_mb: 102.23
-        training_dataset: null
-        task_specialization: null
+        training_dataset: imagenet-1k
+        task_specialization: image-classification
         brainscore_link: https://github.com/brain-score/vision/tree/master/brainscore_vision/models/eBarlow_lmda_001_2
-        huggingface_link: null
+        huggingface_link: https://huggingface.co/timm/resnet50.a1_in1k
         extra_notes: null
         runnable: true