File size: 1,035 Bytes
a78ad5d e664e1c a78ad5d e664e1c 9c57d57 e664e1c a78ad5d 6dc1810 a78ad5d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | {
"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"
} |