{ "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 } }