{ "overall_metrics": { "val_bce_loss": 0.695659875869751, "val_dice_loss": 0.6080484390258789, "val_focal_loss": 0.07589541375637054, "val_mask_loss": 0.6898018717765808, "val_total_loss": 0.6898018717765808, "val_iou": 0.167578125, "val_dice": 0.2576161252133558, "val_f1": 0.2576161252133558, "val_pixel_f1": 0.2576161252133558, "val_auroc": 0.6899952960724477, "val_pixel_auroc": 0.6899952960724477, "val_pixel_acc": 0.4849609375, "val_precision": 0.6, "val_recall": 0.167578125, "val_specificity": 0.9173828125 }, "per_label_metrics": { "Label 0 (Real / Black Mask)": { "iou": 0.0, "dice": 0.0, "f1": 0.0, "pixel_f1": 0.0, "auroc": 1.0, "pixel_auroc": 1.0, "pixel_acc": 0.79345703125, "precision": 0.0, "recall": 0.0, "specificity": 0.79345703125, "samples": 4 }, "Label 1 (Synthetic / White Mask)": { "iou": 0.3098958333333333, "dice": 0.4658045169424121, "f1": 0.4658045169424121, "pixel_f1": 0.4658045169424121, "auroc": 0.4859067289410935, "pixel_auroc": 0.4859067289410935, "pixel_acc": 0.3098958333333333, "precision": 1.0, "recall": 0.3098958333333333, "specificity": 1.0, "samples": 3 }, "Label 2 (Tampered / Partial Mask)": { "iou": 0.24869791666666666, "dice": 0.39291590043544056, "f1": 0.39291590043544056, "pixel_f1": 0.39291590043544056, "auroc": 0.4807442579670654, "pixel_auroc": 0.4807442579670654, "pixel_acc": 0.24869791666666666, "precision": 1.0, "recall": 0.24869791666666666, "specificity": 1.0, "samples": 3 } }, "curves_plot_path": "/tmp/tmpjx19fv4p/out/RUN/test_smoke/reports/training_curves.png", "training_history": [ { "epoch": 1, "lr": 0.001, "train_loss": 0.6793206930160522, "train_mask_loss": 0.6793206930160522, "train_aux_loss": 0.0, "train_iou": 0.28662109375, "val_loss": 0.6806302070617676, "val_mask_loss": 0.6806302070617676, "val_aux_loss": 0.0, "val_iou": 0.6, "val_dice": 0.6, "val_f1": 0.6, "val_pixel_f1": 0.6, "val_auroc": 0.794720982236322, "val_pixel_auroc": 0.794720982236322, "val_pixel_acc": 0.6, "val_precision": 0.6, "val_recall": 0.6, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6314384937286377, "dice_loss": 0.6555118560791016, "focal_loss": 0.07169104367494583, "mask_loss": 0.6793206930160522, "total_loss": 0.6793206930160522, "iou": 0.28662109375 }, "val_metrics": { "val_bce_loss": 0.6884124279022217, "val_dice_loss": 0.5922265648841858, "val_focal_loss": 0.08062143623828888, "val_mask_loss": 0.6806302070617676, "val_total_loss": 0.6806302070617676, "val_iou": 0.6, "val_dice": 0.6, "val_f1": 0.6, "val_pixel_f1": 0.6, "val_auroc": 0.794720982236322, "val_pixel_auroc": 0.794720982236322, "val_pixel_acc": 0.6, "val_precision": 0.6, "val_recall": 0.6, "val_specificity": 0.6 } }, { "epoch": 2, "lr": 0.00025075000000000016, "train_loss": 0.8813323378562927, "train_mask_loss": 0.8813323378562927, "train_aux_loss": 0.0, "train_iou": 0.03301963955163956, "val_loss": 0.9076291473933629, "val_mask_loss": 0.9076291473933629, "val_aux_loss": 0.0, "val_iou": 0.012861883013735859, "val_dice": 0.02538861772258759, "val_f1": 0.02538861772258759, "val_pixel_f1": 0.02538861772258759, "val_auroc": 0.5057031387165009, "val_pixel_auroc": 0.5057031387165009, "val_pixel_acc": 0.014277064732142858, "val_precision": 0.012862112759037647, "val_recall": 0.9985632349484365, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.667902410030365, "dice_loss": 0.9660403728485107, "focal_loss": 0.12872181832790375, "mask_loss": 0.8813323378562927, "total_loss": 0.8813323378562927, "iou": 0.03301963955163956 }, "val_metrics": { "val_bce_loss": 0.7069808265822274, "val_dice_loss": 0.9730980448041643, "val_focal_loss": 0.13517941987514495, "val_mask_loss": 0.9076291473933629, "val_total_loss": 0.9076291473933629, "val_iou": 0.012861883013735859, "val_dice": 0.02538861772258759, "val_f1": 0.02538861772258759, "val_pixel_f1": 0.02538861772258759, "val_auroc": 0.5057031387165009, "val_pixel_auroc": 0.5057031387165009, "val_pixel_acc": 0.014277064732142858, "val_precision": 0.012862112759037647, "val_recall": 0.9985632349484365, "val_specificity": 0.0014522596322735358 } }, { "epoch": 3, "lr": 0.00014730016279731955, "train_loss": 0.6154854893684387, "train_mask_loss": 0.6154854893684387, "train_aux_loss": 0.0, "train_iou": 0.4482421875, "val_loss": 0.6858816146850586, "val_mask_loss": 0.6858816146850586, "val_aux_loss": 0.0, "val_iou": 0.17822265625, "val_dice": 0.21471371091138458, "val_f1": 0.21471371091138458, "val_pixel_f1": 0.21471371091138458, "val_auroc": 0.6987459122174187, "val_pixel_auroc": 0.6987459122174187, "val_pixel_acc": 0.416015625, "val_precision": 0.6, "val_recall": 0.17822265625, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.5453305244445801, "dice_loss": 0.6379597187042236, "focal_loss": 0.047680776566267014, "mask_loss": 0.6154854893684387, "total_loss": 0.6154854893684387, "iou": 0.4482421875 }, "val_metrics": { "val_bce_loss": 0.6939834356307983, "val_dice_loss": 0.6000985503196716, "val_focal_loss": 0.07768132537603378, "val_mask_loss": 0.6858816146850586, "val_total_loss": 0.6858816146850586, "val_iou": 0.17822265625, "val_dice": 0.21471371091138458, "val_f1": 0.21471371091138458, "val_pixel_f1": 0.21471371091138458, "val_auroc": 0.6987459122174187, "val_pixel_auroc": 0.6987459122174187, "val_pixel_acc": 0.416015625, "val_precision": 0.6, "val_recall": 0.17822265625, "val_specificity": 0.83779296875 } }, { "epoch": 4, "lr": 0.001, "train_loss": 0.7411096692085266, "train_mask_loss": 0.7411096692085266, "train_aux_loss": 0.0, "train_iou": 0.17578125, "val_loss": 0.6793217658996582, "val_mask_loss": 0.6793217658996582, "val_aux_loss": 0.0, "val_iou": 0.6, "val_dice": 0.6, "val_f1": 0.6, "val_pixel_f1": 0.6, "val_auroc": 0.79323464758927, "val_pixel_auroc": 0.79323464758927, "val_pixel_acc": 0.6, "val_precision": 0.6, "val_recall": 0.6, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.7112534642219543, "dice_loss": 0.673093318939209, "focal_loss": 0.09787261486053467, "mask_loss": 0.7411096692085266, "total_loss": 0.7411096692085266, "iou": 0.17578125 }, "val_metrics": { "val_bce_loss": 0.6869940757751465, "val_dice_loss": 0.590221107006073, "val_focal_loss": 0.08142829686403275, "val_mask_loss": 0.6793217658996582, "val_total_loss": 0.6793217658996582, "val_iou": 0.6, "val_dice": 0.6, "val_f1": 0.6, "val_pixel_f1": 0.6, "val_auroc": 0.79323464758927, "val_pixel_auroc": 0.79323464758927, "val_pixel_acc": 0.6, "val_precision": 0.6, "val_recall": 0.6, "val_specificity": 0.6 } }, { "epoch": 5, "lr": 6.792031080967287e-05, "train_loss": 0.6793053150177002, "train_mask_loss": 0.6793053150177002, "train_aux_loss": 0.0, "train_iou": 0.28662109375, "val_loss": 0.6815941333770752, "val_mask_loss": 0.6815941333770752, "val_aux_loss": 0.0, "val_iou": 0.6, "val_dice": 0.6, "val_f1": 0.6, "val_pixel_f1": 0.6, "val_auroc": 0.7958538320672233, "val_pixel_auroc": 0.7958538320672233, "val_pixel_acc": 0.6, "val_precision": 0.6, "val_recall": 0.6, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6314182281494141, "dice_loss": 0.6555073857307434, "focal_loss": 0.07168502360582352, "mask_loss": 0.6793053150177002, "total_loss": 0.6793053150177002, "iou": 0.28662109375 }, "val_metrics": { "val_bce_loss": 0.6894435286521912, "val_dice_loss": 0.5937215089797974, "val_focal_loss": 0.08002322912216187, "val_mask_loss": 0.6815941333770752, "val_total_loss": 0.6815941333770752, "val_iou": 0.6, "val_dice": 0.6, "val_f1": 0.6, "val_pixel_f1": 0.6, "val_auroc": 0.7958538320672233, "val_pixel_auroc": 0.7958538320672233, "val_pixel_acc": 0.6, "val_precision": 0.6, "val_recall": 0.6, "val_specificity": 0.6 } }, { "epoch": 6, "lr": 5.046605048274169e-05, "train_loss": 0.6754629611968994, "train_mask_loss": 0.6754629611968994, "train_aux_loss": 0.0, "train_iou": 0.294921875, "val_loss": 0.6836269497871399, "val_mask_loss": 0.6836269497871399, "val_aux_loss": 0.0, "val_iou": 0.58134765625, "val_dice": 0.5904461779109831, "val_f1": 0.5904461779109831, "val_pixel_f1": 0.5904461779109831, "val_auroc": 0.7981927730987082, "val_pixel_auroc": 0.7981927730987082, "val_pixel_acc": 0.591015625, "val_precision": 0.6, "val_recall": 0.58134765625, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.626341700553894, "dice_loss": 0.654387354850769, "focal_loss": 0.07019682228565216, "mask_loss": 0.6754629611968994, "total_loss": 0.6754629611968994, "iou": 0.294921875 }, "val_metrics": { "val_bce_loss": 0.6916059851646423, "val_dice_loss": 0.5967949628829956, "val_focal_loss": 0.07885297387838364, "val_mask_loss": 0.6836269497871399, "val_total_loss": 0.6836269497871399, "val_iou": 0.58134765625, "val_dice": 0.5904461779109831, "val_f1": 0.5904461779109831, "val_pixel_f1": 0.5904461779109831, "val_auroc": 0.7981927730987082, "val_pixel_auroc": 0.7981927730987082, "val_pixel_acc": 0.591015625, "val_precision": 0.6, "val_recall": 0.58134765625, "val_specificity": 0.60966796875 } }, { "epoch": 7, "lr": 3.902217351061228e-05, "train_loss": 0.6724141836166382, "train_mask_loss": 0.6724141836166382, "train_aux_loss": 0.0, "train_iou": 0.2978515625, "val_loss": 0.6857991218566895, "val_mask_loss": 0.6857991218566895, "val_aux_loss": 0.0, "val_iou": 0.090234375, "val_dice": 0.14926459675822704, "val_f1": 0.14926459675822704, "val_pixel_f1": 0.14926459675822704, "val_auroc": 0.6492062145465752, "val_pixel_auroc": 0.6492062145465752, "val_pixel_acc": 0.37578125, "val_precision": 0.6, "val_recall": 0.090234375, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.622309148311615, "dice_loss": 0.65350341796875, "focal_loss": 0.06901578605175018, "mask_loss": 0.6724141836166382, "total_loss": 0.6724141836166382, "iou": 0.2978515625 }, "val_metrics": { "val_bce_loss": 0.6939465403556824, "val_dice_loss": 0.5997651815414429, "val_focal_loss": 0.07788663357496262, "val_mask_loss": 0.6857991218566895, "val_total_loss": 0.6857991218566895, "val_iou": 0.090234375, "val_dice": 0.14926459675822704, "val_f1": 0.14926459675822704, "val_pixel_f1": 0.14926459675822704, "val_auroc": 0.6492062145465752, "val_pixel_auroc": 0.6492062145465752, "val_pixel_acc": 0.37578125, "val_precision": 0.6, "val_recall": 0.090234375, "val_specificity": 0.885546875 } }, { "epoch": 8, "lr": 3.11235359174388e-05, "train_loss": 0.6699907183647156, "train_mask_loss": 0.6699907183647156, "train_aux_loss": 0.0, "train_iou": 0.3037109375, "val_loss": 0.6877641677856445, "val_mask_loss": 0.6877641677856445, "val_aux_loss": 0.0, "val_iou": 0.0185546875, "val_dice": 0.0359900724841848, "val_f1": 0.0359900724841848, "val_pixel_f1": 0.0359900724841848, "val_auroc": 0.6962018286227248, "val_pixel_auroc": 0.6962018286227248, "val_pixel_acc": 0.4041015625, "val_precision": 0.6, "val_recall": 0.0185546875, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.619099497795105, "dice_loss": 0.6528054475784302, "focal_loss": 0.0680764690041542, "mask_loss": 0.6699907183647156, "total_loss": 0.6699907183647156, "iou": 0.3037109375 }, "val_metrics": { "val_bce_loss": 0.6960307955741882, "val_dice_loss": 0.6024684309959412, "val_focal_loss": 0.07702907174825668, "val_mask_loss": 0.6877641677856445, "val_total_loss": 0.6877641677856445, "val_iou": 0.0185546875, "val_dice": 0.0359900724841848, "val_f1": 0.0359900724841848, "val_pixel_f1": 0.0359900724841848, "val_auroc": 0.6962018286227248, "val_pixel_auroc": 0.6962018286227248, "val_pixel_acc": 0.4041015625, "val_precision": 0.6, "val_recall": 0.0185546875, "val_specificity": 0.985546875 } }, { "epoch": 9, "lr": 2.5447270110570814e-05, "train_loss": 0.6680322289466858, "train_mask_loss": 0.6680322289466858, "train_aux_loss": 0.0, "train_iou": 0.3056640625, "val_loss": 0.6898523569107056, "val_mask_loss": 0.6898523569107056, "val_aux_loss": 0.0, "val_iou": 0.0150390625, "val_dice": 0.02921842235873538, "val_f1": 0.02921842235873538, "val_pixel_f1": 0.02921842235873538, "val_auroc": 0.6933273388509406, "val_pixel_auroc": 0.6933273388509406, "val_pixel_acc": 0.403515625, "val_precision": 0.6, "val_recall": 0.0150390625, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6165049076080322, "dice_loss": 0.6522399187088013, "focal_loss": 0.0673196092247963, "mask_loss": 0.6680322289466858, "total_loss": 0.6680322289466858, "iou": 0.3056640625 }, "val_metrics": { "val_bce_loss": 0.6982902884483337, "val_dice_loss": 0.6050440073013306, "val_focal_loss": 0.0763704776763916, "val_mask_loss": 0.6898523569107056, "val_total_loss": 0.6898523569107056, "val_iou": 0.0150390625, "val_dice": 0.02921842235873538, "val_f1": 0.02921842235873538, "val_pixel_f1": 0.02921842235873538, "val_auroc": 0.6933273388509406, "val_pixel_auroc": 0.6933273388509406, "val_pixel_acc": 0.403515625, "val_precision": 0.6, "val_recall": 0.0150390625, "val_specificity": 0.9884765625 } }, { "epoch": 10, "lr": 2.1233259679558623e-05, "train_loss": 0.6664241552352905, "train_mask_loss": 0.6664241552352905, "train_aux_loss": 0.0, "train_iou": 0.310546875, "val_loss": 0.69171541929245, "val_mask_loss": 0.69171541929245, "val_aux_loss": 0.0, "val_iou": 0.01416015625, "val_dice": 0.02753499914319153, "val_f1": 0.02753499914319153, "val_pixel_f1": 0.02753499914319153, "val_auroc": 0.6905683996243169, "val_pixel_auroc": 0.6905683996243169, "val_pixel_acc": 0.40234375, "val_precision": 0.6, "val_recall": 0.01416015625, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6143742203712463, "dice_loss": 0.6517773866653442, "focal_loss": 0.06669659167528152, "mask_loss": 0.6664241552352905, "total_loss": 0.6664241552352905, "iou": 0.310546875 }, "val_metrics": { "val_bce_loss": 0.7002308964729309, "val_dice_loss": 0.6075311899185181, "val_focal_loss": 0.07566872984170914, "val_mask_loss": 0.69171541929245, "val_total_loss": 0.69171541929245, "val_iou": 0.01416015625, "val_dice": 0.02753499914319153, "val_f1": 0.02753499914319153, "val_pixel_f1": 0.02753499914319153, "val_auroc": 0.6905683996243169, "val_pixel_auroc": 0.6905683996243169, "val_pixel_acc": 0.40234375, "val_precision": 0.6, "val_recall": 0.01416015625, "val_specificity": 0.98818359375 } }, { "epoch": 11, "lr": 1.8020049768610437e-05, "train_loss": 0.6650810241699219, "train_mask_loss": 0.6650810241699219, "train_aux_loss": 0.0, "train_iou": 0.31298828125, "val_loss": 0.6928539276123047, "val_mask_loss": 0.6928539276123047, "val_aux_loss": 0.0, "val_iou": 0.01513671875, "val_dice": 0.029402246184059944, "val_f1": 0.029402246184059944, "val_pixel_f1": 0.029402246184059944, "val_auroc": 0.688223364163423, "val_pixel_auroc": 0.688223364163423, "val_pixel_acc": 0.40302734375, "val_precision": 0.6, "val_recall": 0.01513671875, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6125943660736084, "dice_loss": 0.6513910293579102, "focal_loss": 0.0661766305565834, "mask_loss": 0.6650810241699219, "total_loss": 0.6650810241699219, "iou": 0.31298828125 }, "val_metrics": { "val_bce_loss": 0.7012234926223755, "val_dice_loss": 0.6096563339233398, "val_focal_loss": 0.07482807338237762, "val_mask_loss": 0.6928539276123047, "val_total_loss": 0.6928539276123047, "val_iou": 0.01513671875, "val_dice": 0.029402246184059944, "val_f1": 0.029402246184059944, "val_pixel_f1": 0.029402246184059944, "val_auroc": 0.688223364163423, "val_pixel_auroc": 0.688223364163423, "val_pixel_acc": 0.40302734375, "val_precision": 0.6, "val_recall": 0.01513671875, "val_specificity": 0.987890625 } }, { "epoch": 12, "lr": 1.551456219568702e-05, "train_loss": 0.6639510989189148, "train_mask_loss": 0.6639510989189148, "train_aux_loss": 0.0, "train_iou": 0.318359375, "val_loss": 0.6950792074203491, "val_mask_loss": 0.6950792074203491, "val_aux_loss": 0.0, "val_iou": 0.0158203125, "val_dice": 0.030690802899470265, "val_f1": 0.030690802899470265, "val_pixel_f1": 0.030690802899470265, "val_auroc": 0.6857479494588915, "val_pixel_auroc": 0.6857479494588915, "val_pixel_acc": 0.4033203125, "val_precision": 0.6, "val_recall": 0.0158203125, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6110975742340088, "dice_loss": 0.6510665416717529, "focal_loss": 0.06573813408613205, "mask_loss": 0.6639510989189148, "total_loss": 0.6639510989189148, "iou": 0.318359375 }, "val_metrics": { "val_bce_loss": 0.7036783695220947, "val_dice_loss": 0.6119162440299988, "val_focal_loss": 0.07456369698047638, "val_mask_loss": 0.6950792074203491, "val_total_loss": 0.6950792074203491, "val_iou": 0.0158203125, "val_dice": 0.030690802899470265, "val_f1": 0.030690802899470265, "val_pixel_f1": 0.030690802899470265, "val_auroc": 0.6857479494588915, "val_pixel_auroc": 0.6857479494588915, "val_pixel_acc": 0.4033203125, "val_precision": 0.6, "val_recall": 0.0158203125, "val_specificity": 0.9875 } }, { "epoch": 13, "lr": 1.3523507865179105e-05, "train_loss": 0.6629775762557983, "train_mask_loss": 0.6629775762557983, "train_aux_loss": 0.0, "train_iou": 0.32080078125, "val_loss": 0.6980546712875366, "val_mask_loss": 0.6980546712875366, "val_aux_loss": 0.0, "val_iou": 0.017578125, "val_dice": 0.034093589016620324, "val_f1": 0.034093589016620324, "val_pixel_f1": 0.034093589016620324, "val_auroc": 0.6831266120745567, "val_pixel_auroc": 0.6831266120745567, "val_pixel_acc": 0.39970703125, "val_precision": 0.6, "val_recall": 0.017578125, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6098082065582275, "dice_loss": 0.6507865190505981, "focal_loss": 0.065360426902771, "mask_loss": 0.6629775762557983, "total_loss": 0.6629775762557983, "iou": 0.32080078125 }, "val_metrics": { "val_bce_loss": 0.7071222066879272, "val_dice_loss": 0.6143227815628052, "val_focal_loss": 0.07466438412666321, "val_mask_loss": 0.6980546712875366, "val_total_loss": 0.6980546712875366, "val_iou": 0.017578125, "val_dice": 0.034093589016620324, "val_f1": 0.034093589016620324, "val_pixel_f1": 0.034093589016620324, "val_auroc": 0.6831266120745567, "val_pixel_auroc": 0.6831266120745567, "val_pixel_acc": 0.39970703125, "val_precision": 0.6, "val_recall": 0.017578125, "val_specificity": 0.98212890625 } }, { "epoch": 14, "lr": 1.191527343346406e-05, "train_loss": 0.6621235609054565, "train_mask_loss": 0.6621235609054565, "train_aux_loss": 0.0, "train_iou": 0.32763671875, "val_loss": 0.7005167007446289, "val_mask_loss": 0.7005167007446289, "val_aux_loss": 0.0, "val_iou": 0.0287109375, "val_dice": 0.0544594886813052, "val_f1": 0.0544594886813052, "val_pixel_f1": 0.0544594886813052, "val_auroc": 0.6808690778765595, "val_pixel_auroc": 0.6808690778765595, "val_pixel_acc": 0.40546875, "val_precision": 0.6, "val_recall": 0.0287109375, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6086774468421936, "dice_loss": 0.6505409479141235, "focal_loss": 0.06502866744995117, "mask_loss": 0.6621235609054565, "total_loss": 0.6621235609054565, "iou": 0.32763671875 }, "val_metrics": { "val_bce_loss": 0.7099169492721558, "val_dice_loss": 0.616407036781311, "val_focal_loss": 0.07470940053462982, "val_mask_loss": 0.7005167007446289, "val_total_loss": 0.7005167007446289, "val_iou": 0.0287109375, "val_dice": 0.0544594886813052, "val_f1": 0.0544594886813052, "val_pixel_f1": 0.0544594886813052, "val_auroc": 0.6808690778765595, "val_pixel_auroc": 0.6808690778765595, "val_pixel_acc": 0.40546875, "val_precision": 0.6, "val_recall": 0.0287109375, "val_specificity": 0.9767578125 } }, { "epoch": 15, "lr": 1.0597752438586402e-05, "train_loss": 0.6613755226135254, "train_mask_loss": 0.6613755226135254, "train_aux_loss": 0.0, "train_iou": 0.330078125, "val_loss": 0.701773464679718, "val_mask_loss": 0.701773464679718, "val_aux_loss": 0.0, "val_iou": 0.03935546875, "val_dice": 0.07281441976440073, "val_f1": 0.07281441976440073, "val_pixel_f1": 0.07281441976440073, "val_auroc": 0.6800669258431299, "val_pixel_auroc": 0.6800669258431299, "val_pixel_acc": 0.40478515625, "val_precision": 0.6, "val_recall": 0.03935546875, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6076864004135132, "dice_loss": 0.6503254175186157, "focal_loss": 0.0647391751408577, "mask_loss": 0.6613755226135254, "total_loss": 0.6613755226135254, "iou": 0.330078125 }, "val_metrics": { "val_bce_loss": 0.7113760113716125, "val_dice_loss": 0.6171571612358093, "val_focal_loss": 0.07501374185085297, "val_mask_loss": 0.701773464679718, "val_total_loss": 0.701773464679718, "val_iou": 0.03935546875, "val_dice": 0.07281441976440073, "val_f1": 0.07281441976440073, "val_pixel_f1": 0.07281441976440073, "val_auroc": 0.6800669258431299, "val_pixel_auroc": 0.6800669258431299, "val_pixel_acc": 0.40478515625, "val_precision": 0.6, "val_recall": 0.03935546875, "val_specificity": 0.9654296875 } }, { "epoch": 16, "lr": 9.50493670789106e-06, "train_loss": 0.6607082486152649, "train_mask_loss": 0.6607082486152649, "train_aux_loss": 0.0, "train_iou": 0.3349609375, "val_loss": 0.6997901797294617, "val_mask_loss": 0.6997901797294617, "val_aux_loss": 0.0, "val_iou": 0.05732421875, "val_dice": 0.10170591155159267, "val_f1": 0.10170591155159267, "val_pixel_f1": 0.10170591155159267, "val_auroc": 0.6815693911223206, "val_pixel_auroc": 0.6815693911223206, "val_pixel_acc": 0.41123046875, "val_precision": 0.6, "val_recall": 0.05732421875, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6068019866943359, "dice_loss": 0.6501330137252808, "focal_loss": 0.06448154151439667, "mask_loss": 0.6607082486152649, "total_loss": 0.6607082486152649, "iou": 0.3349609375 }, "val_metrics": { "val_bce_loss": 0.7088394165039062, "val_dice_loss": 0.6157937049865723, "val_focal_loss": 0.07494723796844482, "val_mask_loss": 0.6997901797294617, "val_total_loss": 0.6997901797294617, "val_iou": 0.05732421875, "val_dice": 0.10170591155159267, "val_f1": 0.10170591155159267, "val_pixel_f1": 0.10170591155159267, "val_auroc": 0.6815693911223206, "val_pixel_auroc": 0.6815693911223206, "val_pixel_acc": 0.41123046875, "val_precision": 0.6, "val_recall": 0.05732421875, "val_specificity": 0.95390625 } }, { "epoch": 17, "lr": 8.588527370402095e-06, "train_loss": 0.6601073741912842, "train_mask_loss": 0.6601073741912842, "train_aux_loss": 0.0, "train_iou": 0.3369140625, "val_loss": 0.6963794231414795, "val_mask_loss": 0.6963794231414795, "val_aux_loss": 0.0, "val_iou": 0.08349609375, "val_dice": 0.1404100559325246, "val_f1": 0.1404100559325246, "val_pixel_f1": 0.1404100559325246, "val_auroc": 0.6843118134245743, "val_pixel_auroc": 0.6843118134245743, "val_pixel_acc": 0.423046875, "val_precision": 0.6, "val_recall": 0.08349609375, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6060049533843994, "dice_loss": 0.6499600410461426, "focal_loss": 0.06424979120492935, "mask_loss": 0.6601073741912842, "total_loss": 0.6601073741912842, "iou": 0.3369140625 }, "val_metrics": { "val_bce_loss": 0.704594075679779, "val_dice_loss": 0.6133010983467102, "val_focal_loss": 0.07486365735530853, "val_mask_loss": 0.6963794231414795, "val_total_loss": 0.6963794231414795, "val_iou": 0.08349609375, "val_dice": 0.1404100559325246, "val_f1": 0.1404100559325246, "val_pixel_f1": 0.1404100559325246, "val_auroc": 0.6843118134245743, "val_pixel_auroc": 0.6843118134245743, "val_pixel_acc": 0.423046875, "val_precision": 0.6, "val_recall": 0.08349609375, "val_specificity": 0.93955078125 } }, { "epoch": 18, "lr": 7.812528950340201e-06, "train_loss": 0.6595633029937744, "train_mask_loss": 0.6595633029937744, "train_aux_loss": 0.0, "train_iou": 0.3388671875, "val_loss": 0.6927188038825989, "val_mask_loss": 0.6927188038825989, "val_aux_loss": 0.0, "val_iou": 0.11240234375, "val_dice": 0.18207771628976493, "val_f1": 0.18207771628976493, "val_pixel_f1": 0.18207771628976493, "val_auroc": 0.6869782303285319, "val_pixel_auroc": 0.6869782303285319, "val_pixel_acc": 0.4474609375, "val_precision": 0.6, "val_recall": 0.11240234375, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.605282723903656, "dice_loss": 0.6498034000396729, "focal_loss": 0.06404046714305878, "mask_loss": 0.6595633029937744, "total_loss": 0.6595633029937744, "iou": 0.3388671875 }, "val_metrics": { "val_bce_loss": 0.6999369859695435, "val_dice_loss": 0.6108706593513489, "val_focal_loss": 0.07462991774082184, "val_mask_loss": 0.6927188038825989, "val_total_loss": 0.6927188038825989, "val_iou": 0.11240234375, "val_dice": 0.18207771628976493, "val_f1": 0.18207771628976493, "val_pixel_f1": 0.18207771628976493, "val_auroc": 0.6869782303285319, "val_pixel_auroc": 0.6869782303285319, "val_pixel_acc": 0.4474609375, "val_precision": 0.6, "val_recall": 0.11240234375, "val_specificity": 0.93505859375 } }, { "epoch": 19, "lr": 7.1496738727287385e-06, "train_loss": 0.6590670347213745, "train_mask_loss": 0.6590670347213745, "train_aux_loss": 0.0, "train_iou": 0.34033203125, "val_loss": 0.6899673938751221, "val_mask_loss": 0.6899673938751221, "val_aux_loss": 0.0, "val_iou": 0.13125, "val_dice": 0.20949789932585375, "val_f1": 0.20949789932585375, "val_pixel_f1": 0.20949789932585375, "val_auroc": 0.6889150603063172, "val_pixel_auroc": 0.6889150603063172, "val_pixel_acc": 0.46123046875, "val_precision": 0.6, "val_recall": 0.13125, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6046238541603088, "dice_loss": 0.6496597528457642, "focal_loss": 0.06385035812854767, "mask_loss": 0.6590670347213745, "total_loss": 0.6590670347213745, "iou": 0.34033203125 }, "val_metrics": { "val_bce_loss": 0.6963634490966797, "val_dice_loss": 0.6090818643569946, "val_focal_loss": 0.07448948174715042, "val_mask_loss": 0.6899673938751221, "val_total_loss": 0.6899673938751221, "val_iou": 0.13125, "val_dice": 0.20949789932585375, "val_f1": 0.20949789932585375, "val_pixel_f1": 0.20949789932585375, "val_auroc": 0.6889150603063172, "val_pixel_auroc": 0.6889150603063172, "val_pixel_acc": 0.46123046875, "val_precision": 0.6, "val_recall": 0.13125, "val_specificity": 0.92998046875 } }, { "epoch": 20, "lr": 6.579002300548303e-06, "train_loss": 0.6586103439331055, "train_mask_loss": 0.6586103439331055, "train_aux_loss": 0.0, "train_iou": 0.3427734375, "val_loss": 0.6887679696083069, "val_mask_loss": 0.6887679696083069, "val_aux_loss": 0.0, "val_iou": 0.14560546875, "val_dice": 0.2294412204840397, "val_f1": 0.2294412204840397, "val_pixel_f1": 0.2294412204840397, "val_auroc": 0.689870517826057, "val_pixel_auroc": 0.689870517826057, "val_pixel_acc": 0.4708984375, "val_precision": 0.6, "val_recall": 0.14560546875, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.604017436504364, "dice_loss": 0.6495270729064941, "focal_loss": 0.06367617100477219, "mask_loss": 0.6586103439331055, "total_loss": 0.6586103439331055, "iou": 0.3427734375 }, "val_metrics": { "val_bce_loss": 0.6946761608123779, "val_dice_loss": 0.6081814765930176, "val_focal_loss": 0.07467827200889587, "val_mask_loss": 0.6887679696083069, "val_total_loss": 0.6887679696083069, "val_iou": 0.14560546875, "val_dice": 0.2294412204840397, "val_f1": 0.2294412204840397, "val_pixel_f1": 0.2294412204840397, "val_auroc": 0.689870517826057, "val_pixel_auroc": 0.689870517826057, "val_pixel_acc": 0.4708984375, "val_precision": 0.6, "val_recall": 0.14560546875, "val_specificity": 0.92529296875 } }, { "epoch": 21, "lr": 6.084189780474125e-06, "train_loss": 0.6581863164901733, "train_mask_loss": 0.6581863164901733, "train_aux_loss": 0.0, "train_iou": 0.34375, "val_loss": 0.6893299221992493, "val_mask_loss": 0.6893299221992493, "val_aux_loss": 0.0, "val_iou": 0.15927734375, "val_dice": 0.2473718957302852, "val_f1": 0.2473718957302852, "val_pixel_f1": 0.2473718957302852, "val_auroc": 0.689975963276811, "val_pixel_auroc": 0.689975963276811, "val_pixel_acc": 0.47900390625, "val_precision": 0.6, "val_recall": 0.15927734375, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6034538745880127, "dice_loss": 0.6494040489196777, "focal_loss": 0.06351464986801147, "mask_loss": 0.6581863164901733, "total_loss": 0.6581863164901733, "iou": 0.34375 }, "val_metrics": { "val_bce_loss": 0.6951814293861389, "val_dice_loss": 0.6080673933029175, "val_focal_loss": 0.07541101425886154, "val_mask_loss": 0.6893299221992493, "val_total_loss": 0.6893299221992493, "val_iou": 0.15927734375, "val_dice": 0.2473718957302852, "val_f1": 0.2473718957302852, "val_pixel_f1": 0.2473718957302852, "val_auroc": 0.689975963276811, "val_pixel_auroc": 0.689975963276811, "val_pixel_acc": 0.47900390625, "val_precision": 0.6, "val_recall": 0.15927734375, "val_specificity": 0.9197265625 } }, { "epoch": 22, "lr": 1e-06, "train_loss": 0.6578466296195984, "train_mask_loss": 0.6578466296195984, "train_aux_loss": 0.0, "train_iou": 0.34423828125, "val_loss": 0.6898018717765808, "val_mask_loss": 0.6898018717765808, "val_aux_loss": 0.0, "val_iou": 0.167578125, "val_dice": 0.2576161252133558, "val_f1": 0.2576161252133558, "val_pixel_f1": 0.2576161252133558, "val_auroc": 0.6899952960724477, "val_pixel_auroc": 0.6899952960724477, "val_pixel_acc": 0.4849609375, "val_precision": 0.6, "val_recall": 0.167578125, "val_aux_accuracy": 0.0, "train_metrics": { "bce_loss": 0.6030022501945496, "dice_loss": 0.6493053436279297, "focal_loss": 0.06338557600975037, "mask_loss": 0.6578466296195984, "total_loss": 0.6578466296195984, "iou": 0.34423828125 }, "val_metrics": { "val_bce_loss": 0.695659875869751, "val_dice_loss": 0.6080484390258789, "val_focal_loss": 0.07589541375637054, "val_mask_loss": 0.6898018717765808, "val_total_loss": 0.6898018717765808, "val_iou": 0.167578125, "val_dice": 0.2576161252133558, "val_f1": 0.2576161252133558, "val_pixel_f1": 0.2576161252133558, "val_auroc": 0.6899952960724477, "val_pixel_auroc": 0.6899952960724477, "val_pixel_acc": 0.4849609375, "val_precision": 0.6, "val_recall": 0.167578125, "val_specificity": 0.9173828125 } } ], "configuration": { "project": { "name": "test_smoke", "seed": 42, "device": "cpu", "output_dir": "/tmp/tmpjx19fv4p/out/RUN/test_smoke", "hub_repo": null, "resume_repo": null }, "data": { "dataset_name": "KhangTruong/IMD2020", "streaming": false, "image_size": [ 32, 32 ], "batch_size": 16, "num_workers": 0, "pin_memory": false, "shuffle_buffer_size": 1000, "train_split": "train", "val_split": "validation", "test_split": "test", "eval_split": "test", "train_samples_per_epoch": 2, "val_samples_per_epoch": -1, "val_samples": 2, "test_samples": 2, "evaluate_on_test": true, "augmentations": { "horizontal_flip": 0.5, "vertical_flip": 0.0, "random_rotate90": 0.0, "color_jitter": 0.0 }, "base_batch_size": 16, "num_gpus": 1, "_batch_size_scaled": true }, "model": { "name": "unet", "in_channels": 3, "out_channels": 1, "features": [ 8, 16 ], "bilinear": true, "dropout": 0.0, "aux_classifier": false, "num_classes": 3, "gradient_checkpointing": false }, "loss": { "mask_loss_type": "combined", "bce_weight": 0.5, "dice_weight": 0.5, "focal_weight": 0.5, "focal_gamma": 2.0, "focal_alpha": 0.25, "aux_loss_type": "cross_entropy", "aux_weight": 0.2 }, "training": { "epochs": 1, "learning_rate": 0.001, "weight_decay": 0.0001, "optimizer": "adamw", "use_8bit_optimizer": false, "check_8bit_compatibility": true, "fallback_to_16bit": true, "fallback_on_unsupported_8bit": true, "scheduler": "cosine", "resume_lr_mode": "auto", "resume_lr": null, "warmup_epochs": 1, "min_lr": 1e-06, "grad_clip_norm": 1.0, "amp": false, "gradient_accumulation_steps": 1, "data_parallel": false, "use_hard_mining": false, "auto_batch_size": true, "empty_cache_per_epoch": true, "save_best": true, "save_latest": true, "checkpoint_period": 3600, "eval_interval": 1, "early_stopping_patience": 2, "early_stopping_metric": "val_iou", "early_stopping_mode": "max", "push_to_hub": true, "hub_repo": "KhangTruong/Testing-model", "hub_version": "v1", "batch_size": 2, "hf_repo": "KhangTruong/Testing-model", "num_gpus": 1 }, "hard_mining": { "enabled": false, "metric": "median", "reset_epochs": 5 }, "bootstrapping": { "enabled": false, "run_bootstrap": false, "epochs": 5, "num_samples": 512, "batch_size": null, "learning_rate": null, "freeze_strategy": "channel_stream", "stream_ratio": 0.5, "release_mode": "restore", "freeze_modules": [], "initialization": "kaiming_normal", "target_score": 0.5, "min_loss_drop": 0.15, "early_stopping": false, "patience": 3 }, "logging": { "log_interval": 1, "log_memory": true, "measure_network": true }, "post_processing": { "enabled": true, "min_area": 64, "fill_holes": true, "max_hole_size": 256, "morphology": "open_close", "morph_kernel_size": 3 } }, "hyperparameters": { "Project Name": "test_smoke", "Dataset": "KhangTruong/IMD2020", "Splits": "Train: 'train' | Val: 'validation' | Test: 'test'", "Streaming": false, "Image Size": "[32, 32]", "Batch Size": 16, "Epoch Limit / Steps": "2 samples/epoch", "Model": "unet (features=[8, 16], bilinear=True)", "Aux Classifier": "False (classes=3)", "Mask Loss": "Combined (BCE: 0.5, Dice: 0.5, Focal: 0.5)", "Aux Loss": "cross_entropy (wt=0.2)", "Post-Processing": "Enabled (min_area=64, fill_holes=True, morph=open_close)", "Optimizer": "adamw (lr=0.001, wd=0.0001)", "Scheduler": "cosine", "Epochs": 1, "AMP": false } }