{
  "device": "cuda",
  "epochs": 25,
  "learning_rate": 0.001,
  "num_classes": 2,
  "class_names": {
    "0": "background",
    "1": "lung"
  },
  "num_train_cases": 13,
  "num_val_cases": 7,
  "best_val_dice": 0.9539202451705933,
  "history": [
    {
      "epoch": 1,
      "train_loss": 0.7965623965630164,
      "val_dice": 0.8137869834899902
    },
    {
      "epoch": 2,
      "train_loss": 0.45559382897156936,
      "val_dice": 0.8940457105636597
    },
    {
      "epoch": 3,
      "train_loss": 0.29859031163729155,
      "val_dice": 0.9218230843544006
    },
    {
      "epoch": 4,
      "train_loss": 0.22275064083246085,
      "val_dice": 0.9315038323402405
    },
    {
      "epoch": 5,
      "train_loss": 0.17803634817783648,
      "val_dice": 0.936585545539856
    },
    {
      "epoch": 6,
      "train_loss": 0.1547177445430022,
      "val_dice": 0.9365377426147461
    },
    {
      "epoch": 7,
      "train_loss": 0.14395424952873817,
      "val_dice": 0.9384771585464478
    },
    {
      "epoch": 8,
      "train_loss": 0.13710603920313028,
      "val_dice": 0.9324427247047424
    },
    {
      "epoch": 9,
      "train_loss": 0.1219253597351221,
      "val_dice": 0.9358949661254883
    },
    {
      "epoch": 10,
      "train_loss": 0.1086675117795284,
      "val_dice": 0.9450822472572327
    },
    {
      "epoch": 11,
      "train_loss": 0.09993432691464058,
      "val_dice": 0.9439553022384644
    },
    {
      "epoch": 12,
      "train_loss": 0.09189996524499013,
      "val_dice": 0.9456437230110168
    },
    {
      "epoch": 13,
      "train_loss": 0.08811248208467777,
      "val_dice": 0.9494844079017639
    },
    {
      "epoch": 14,
      "train_loss": 0.08259457865586647,
      "val_dice": 0.9439782500267029
    },
    {
      "epoch": 15,
      "train_loss": 0.08551208904156318,
      "val_dice": 0.9461687207221985
    },
    {
      "epoch": 16,
      "train_loss": 0.08220311655448033,
      "val_dice": 0.9465957283973694
    },
    {
      "epoch": 17,
      "train_loss": 0.07529287556043038,
      "val_dice": 0.94902503490448
    },
    {
      "epoch": 18,
      "train_loss": 0.07014285744382785,
      "val_dice": 0.9508883357048035
    },
    {
      "epoch": 19,
      "train_loss": 0.07010523659678605,
      "val_dice": 0.9494710564613342
    },
    {
      "epoch": 20,
      "train_loss": 0.06804308151969543,
      "val_dice": 0.9491676688194275
    },
    {
      "epoch": 21,
      "train_loss": 0.06551067588421014,
      "val_dice": 0.9507818818092346
    },
    {
      "epoch": 22,
      "train_loss": 0.06288302517854251,
      "val_dice": 0.9500231146812439
    },
    {
      "epoch": 23,
      "train_loss": 0.06616245955228806,
      "val_dice": 0.9527921676635742
    },
    {
      "epoch": 24,
      "train_loss": 0.06158428237988399,
      "val_dice": 0.9526895880699158
    },
    {
      "epoch": 25,
      "train_loss": 0.0577649657542889,
      "val_dice": 0.9539202451705933
    }
  ],
  "checkpoint": "outputs/zenodo_lung_full_e25/checkpoints/best_model.pt"
}