| { | |
| "data_root": "/content/data/Dataset_BUSI_with_GT", | |
| "img_size": 256, | |
| "n_classes": 1, | |
| "exclude_normal": true, | |
| "epochs": 120, | |
| "warmup_epochs": 0, | |
| "bs": 2, | |
| "n_folds": 1, | |
| "seed": 42, | |
| "num_workers": 2, | |
| "pin_memory": true, | |
| "use_amp": false, | |
| "grad_clip": 0.5, | |
| "accum_steps": 1, | |
| "w_dice": 1.0, | |
| "w_bce": 0.5, | |
| "w_boundary": 0.8, | |
| "w_stage": 0.3, | |
| "w_cls": 0.3, | |
| "w_hd": 1.0, | |
| "sal_freeze_epochs": 0, | |
| "sal_lambda_init": 0.1, | |
| "sal_decay_end": 20, | |
| "backbone": "convnext_tiny", | |
| "pretrained": true, | |
| "scale": "small", | |
| "dims": [ | |
| 64, | |
| 128, | |
| 256, | |
| 512 | |
| ], | |
| "mb": [ | |
| 2, | |
| 2, | |
| 3 | |
| ], | |
| "dp": 0.1, | |
| "lr": 0.0001, | |
| "note": "~24M \u2014 T4 15GB PRIMARY", | |
| "infer_size": 256, | |
| "sam_rho": 0.05, | |
| "ema_decay": 0.999, | |
| "save_dir": "/content/checkpoints", | |
| "log_interval": 10, | |
| "hf_repo": "Swarno-Coder/SonaMamba", | |
| "hf_push": true, | |
| "accum_bs": 2, | |
| "best_metrics": { | |
| "dsc": 0.16328166262246668, | |
| "iou": 0.09806505093315186, | |
| "sens": 0.9997425597447616, | |
| "spec": 0.061594799877359316, | |
| "prec": 0.09807573254399288, | |
| "auc": 0.529585798816568 | |
| } | |
| } |