src/model_config.json

{
  "models": {
    "vit": {
      "model_name": "vit_base_patch16_224.augreg_in21k_ft_in1k",
      "architecture": "ViT-B/16",
      "num_classes": 23,
      "pretrained_source": "AugReg ImageNet-21k fine-tuned on ImageNet-1k",
      "checkpoint_file": "vit_b16_gravityspy_o3.pt",
      "input_size": 224,
      "normalization": {
        "mean": [0.485, 0.456, 0.406],
        "std": [0.229, 0.224, 0.225]
      }
    },
    "cnn": {
      "model_name": "resnetv2_50x1_bit.goog_in21k_ft_in1k",
      "architecture": "ResNet-50v2 BiT",
      "num_classes": 23,
      "pretrained_source": "ImageNet-21k fine-tuned on ImageNet-1k",
      "checkpoint_file": "resnet50v2_gravityspy_o3.pt",
      "input_size": 224,
      "normalization": {
        "mean": [0.485, 0.456, 0.406],
        "std": [0.229, 0.224, 0.225]
      }
    }
  },
  "training": {
    "dataset": "Gravity Spy O3",
    "num_classes": 23,
    "split": "temporal 70/15/15% with 60s gap enforcement",
    "train_samples": 227943,
    "seed": 42
  }
}