{
  "best_metric": 0.7789129817933231,
  "checkpoint_source": "saved/checkpoints/four_ch_morph/lodo_aptos_convnext_tiny_best.pt",
  "class_names": [
    "No DR",
    "Mild",
    "Moderate",
    "Severe",
    "Proliferative DR"
  ],
  "ema_state_dict_present": true,
  "epoch": 12,
  "global_step": 13742,
  "phase_name": "phase2_full_training",
  "primary_weights": "lodo_aptos_convnext_tiny_best_ema.safetensors",
  "primary_weights_kind": "ema",
  "secondary_weights": "lodo_aptos_convnext_tiny_best.safetensors",
  "secondary_weights_kind": "raw_model_state_dict",
  "training_config": {
    "arch": "convnext_tiny",
    "cbam_num_stages": 2,
    "channels_last": true,
    "dropout": 0.4,
    "ema_decay": 0.999,
    "head_hidden_dim": 512,
    "img_size": 384,
    "in_chans": 4,
    "loss_type": "corn",
    "num_thresholds": 4,
    "phase_batches": {
      "phase1_frozen": 64,
      "phase2_full_training": 24
    },
    "use_cbam": true,
    "use_class_weighting": true,
    "use_mixup": true
  }
}