training_config.json

{
  "dataset_index": "outputs/model/image_type_classifier_dataset_index.csv",
  "output_dir": "models/image_type_classifier_v1",
  "epochs": 10,
  "batch_size": 32,
  "lr": 0.0001,
  "weight_decay": 0.01,
  "backbone": "efficientnet_b0",
  "image_size": 224,
  "seed": 42,
  "num_workers": 0,
  "augmentation": "mild_clinical",
  "max_train_batches": 60,
  "max_val_batches": 60,
  "class_names": [
    "clinical_macroscopic",
    "dermoscopic"
  ],
  "class_to_idx": {
    "clinical_macroscopic": 0,
    "dermoscopic": 1
  },
  "device": "mps",
  "trainable_params": 4010110,
  "train_examples": 19324,
  "val_examples": 4153
}