Results epoch 51

baseline_results.json

@@ -1,56 +1,33 @@
 {
     "experiment": "Baseline Original 3D U-Net",
-    "status": "completed",
-    "completed_epochs": 50,
+    "status": "training",
+    "last_completed_epoch": 51,
+    "next_epoch": 52,
     "best_epoch": 45,
     "best_validation_dice": 0.7721250271246268,
-    "best_epoch_metrics": {
-        "epoch": 45,
-        "learning_rate": 3.5111757055874303e-06,
-        "train_loss": 0.3746598991407018,
-        "val_loss": 0.476624794908472,
-        "train_dice": 0.84932605254679,
-        "val_dice": 0.7721250271246268,
-        "train_dice_macro_all": 0.8824622418971645,
-        "val_dice_macro_all": 0.8239281408112812,
-        "train_tumor_dice": 0.9395233504399889,
-        "val_tumor_dice": 0.9241284054564372,
-        "train_iou": 0.7384798154511153,
-        "val_iou": 0.6346612539007251,
-        "train_tumor_iou": 0.8859444013498974,
-        "val_tumor_iou": 0.8589578999420441,
-        "train_precision": 0.9468324232065657,
-        "val_precision": 0.9469938214247372,
-        "train_recall": 0.9323262580137129,
-        "val_recall": 0.9023411402697056,
-        "train_fnr": 0.06767374198628705,
-        "val_fnr": 0.09765885973029444,
-        "train_fpr": 0.015851784619549208,
-        "val_fpr": 0.014180152025839282,
-        "train_TP": 16700075,
-        "train_TN": 58220310,
-        "train_FP": 937761,
-        "train_FN": 1212190,
-        "val_TP": 3837129,
-        "val_TN": 14931465,
-        "val_FP": 214776,
-        "val_FN": 415286,
-        "val_dice_background": 0.9793374818712444,
-        "val_dice_ncr_net": 0.6698967303134118,
-        "val_dice_edema": 0.801320658082598,
-        "val_dice_enhancing": 0.8451576929778707,
-        "val_iou_background": 0.9595115569313988,
-        "val_iou_ncr_net": 0.503642646086615,
-        "val_iou_edema": 0.6685029349056847,
-        "val_iou_enhancing": 0.7318381807098756,
-        "epoch_seconds": 398.597907
-    },
-    "total_training_seconds": 14417.651626,
-    "files": {
-        "best_model": "/kaggle/working/baseline_3dunet_results/baseline_best_model.pth",
-        "latest_checkpoint": "/kaggle/working/baseline_3dunet_results/baseline_latest_checkpoint.pth",
-        "history_csv": "/kaggle/working/baseline_3dunet_results/baseline_history.csv",
-        "results_json": "/kaggle/working/baseline_3dunet_results/baseline_results.json"
+    "config": {
+        "num_epochs": 50,
+        "batch_size": 1,
+        "num_classes": 4,
+        "model": "Original3DUNet",
+        "loss": "Dice + Cross Entropy",
+        "patch_size": [
+            64,
+            64,
+            64
+        ],
+        "modalities": [
+            "flair",
+            "t1",
+            "t1ce",
+            "t2"
+        ],
+        "train_patients": 294,
+        "validation_patients": 74,
+        "sampling": "Tumor-centered random sampling",
+        "dataset": "BraTS2020",
+        "excluded_patient": "BraTS20_Training_355",
+        "hf_repository": "Reduanul1997/RepositoryBaseLine3DuNet"
     },
     "history": [
         {
@@ -2102,6 +2079,47 @@
             "val_iou_edema": 0.673685773055962,
             "val_iou_enhancing": 0.7137133904088354,
             "epoch_seconds": 416.356897
+        },
+        {
+            "epoch": 51,
+            "learning_rate": 0.0,
+            "train_loss": 0.36533703647401866,
+            "val_loss": 0.5158799701445812,
+            "train_dice": 0.8550863165959585,
+            "val_dice": 0.7510871968253165,
+            "train_dice_macro_all": 0.8869575969585125,
+            "val_dice_macro_all": 0.8075737551613263,
+            "train_tumor_dice": 0.9411262246091566,
+            "val_tumor_dice": 0.9182617356533097,
+            "train_iou": 0.7470855382567466,
+            "val_iou": 0.6101796263371154,
+            "train_tumor_iou": 0.8887992567969448,
+            "val_tumor_iou": 0.8488760783625314,
+            "train_precision": 0.9481925927002732,
+            "val_precision": 0.9557556434639924,
+            "train_recall": 0.9341644010657464,
+            "val_recall": 0.8835985225479241,
+            "train_fnr": 0.06583559893425366,
+            "val_fnr": 0.11640147745207587,
+            "train_fpr": 0.015256047430495636,
+            "val_fpr": 0.012077077763160275,
+            "train_TP": 16567571,
+            "train_TN": 58429939,
+            "train_FP": 905220,
+            "train_FN": 1167606,
+            "val_TP": 3907217,
+            "val_TN": 14795844,
+            "val_FP": 180875,
+            "val_FN": 514720,
+            "val_dice_background": 0.9770334301693553,
+            "val_dice_ncr_net": 0.6192431222744172,
+            "val_dice_edema": 0.7988443883196326,
+            "val_dice_enhancing": 0.8351740798819,
+            "val_iou_background": 0.9550981028941211,
+            "val_iou_ncr_net": 0.44848092539973433,
+            "val_iou_edema": 0.665063194603139,
+            "val_iou_enhancing": 0.716994759008473,
+            "epoch_seconds": 933.95946
         }
     ]
 }