{
  "dataset": {
    "raw_dir": "/content/drive/MyDrive/nsl_raw",
    "num_raw_images": 4815,
    "num_valid_labeled_images": 4812,
    "num_classes": 137,
    "min_total_per_class": 1,
    "dropped_classes": [],
    "split_protocol": "official source test split; validation split from source train by class",
    "split_counts": {
      "train": 3581,
      "val": 636,
      "test": 595
    },
    "classes_per_split": {
      "test": 136,
      "train": 137,
      "val": 136
    },
    "duplicate_hashes_crossing_splits": 0
  },
  "vit_classification": {
    "model": "vit_base_patch16_224",
    "input_mode": "crop",
    "epochs_requested": 40,
    "head_warmup_epochs": 3,
    "best_checkpoint": "/content/nsl_vit_fixed_work/runs/vit_b16_nsl_fixed_best.pt",
    "params": 85904009,
    "val": {
      "loss": 0.8947618052644549,
      "top1": 0.9261006289308176,
      "top5": 0.9638364779874213,
      "macro_f1": 0.9024900968879636,
      "weighted_f1": 0.9203976265041456
    },
    "test": {
      "loss": 0.8799869286913832,
      "top1": 0.9210084033613445,
      "top5": 0.9697478991596639,
      "macro_f1": 0.9066332142953599,
      "weighted_f1": 0.9183226019280669
    }
  }
}