feat(release): final training report and checkpoints for KhangTruong/Testing-model (v1)
1d7c979 verified | { | |
| "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 | |
| } | |
| } |