{
  "model_type": "unet3plus",
  "backbone": "efficientnet",
  "img_size": 256,
  "in_channels": 3,
  "out_channels": 1,
  "inter_ch": 64,
  "auto_map": {
    "AutoConfig": "modeling_unet3plus.UNet3PlusConfig",
    "AutoModel": "modeling_unet3plus.UNet3PlusForSegmentation"
  },
  "training": {
    "loss": "dice_focal",
    "focal_gamma": 1.1217038657488427,
    "focal_dice_weight": 0.3012325073214737,
    "lr": 0.0017935009239058379,
    "weight_decay": 2.5092757604508502e-06,
    "warmup_ratio": 0.12072068963857166,
    "scheduler": "cosine_with_restarts",
    "batch_size": 64,
    "epochs": 55,
    "optimiser": "AdamW"
  },
  "test_metrics": {
    "dice": 0.9233709573745728,
    "iou": 0.8576500415802002,
    "f1": 0.9233709475438584,
    "precision": 0.9474478438948487,
    "recall": 0.9004874267475651,
    "accuracy": 0.9744630432128907,
    "loss": 0.09141451492905617
  },
  "dataset": "andreribeiro87/kvasir-seg-augmented",
  "hpo": "Optuna MedianPruner, 60 trials (28 completed, 32 pruned), best eval_loss=0.0537"
}