File size: 1,121 Bytes
21a69cf 6f6b63d 21a69cf f42a44f 21a69cf 6f6b63d 21a69cf 6f6b63d 21a69cf 6f6b63d 21a69cf 6f6b63d 21a69cf 6f6b63d 21a69cf 6f6b63d 21a69cf d63a857 21a69cf | 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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | {
"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
}
} |