| { |
| "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" |
| } |
