You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.6 KiB
JSON
53 lines
1.6 KiB
JSON
{
|
|
"backbone_type" : "nontransformer",
|
|
"task": "segmentation",
|
|
"n_classes" : 3,
|
|
"n_epochs" : 50,
|
|
"input_height" : 448,
|
|
"input_width" : 448,
|
|
"weight_decay" : 1e-4,
|
|
"n_batch" : 4,
|
|
"learning_rate": 5e-5,
|
|
"patches" : false,
|
|
"pretraining" : true,
|
|
"augmentation" : true,
|
|
"flip_aug" : false,
|
|
"blur_aug" : false,
|
|
"scaling" : true,
|
|
"degrading": true,
|
|
"brightening": false,
|
|
"binarization" : false,
|
|
"scaling_bluring" : false,
|
|
"scaling_binarization" : false,
|
|
"scaling_flip" : false,
|
|
"rotation": false,
|
|
"rotation_not_90": false,
|
|
"blur_k" : ["blur","guass","median"],
|
|
"scales" : [0.6, 0.7, 0.8, 0.9],
|
|
"brightness" : [1.3, 1.5, 1.7, 2],
|
|
"degrade_scales" : [0.2, 0.4],
|
|
"flip_index" : [0, 1, -1],
|
|
"thetha" : [10, -10],
|
|
"continue_training": false,
|
|
"index_start" : 0,
|
|
"dir_of_start_model" : " ",
|
|
"weighted_loss": false,
|
|
"is_loss_soft_dice": true,
|
|
"data_is_provided": true,
|
|
"dir_train": "/home/incognito/sbb_pixelwise_segmentation/dataset/sam_41_mss/dir_train/train",
|
|
"dir_eval": "/home/incognito/sbb_pixelwise_segmentation/dataset/sam_41_mss/dir_train/eval",
|
|
"dir_output": "runs/sam_41_mss_npt_448x448",
|
|
"reduce_lr_enabled": true,
|
|
"reduce_lr_monitor": "val_loss",
|
|
"reduce_lr_factor": 0.5,
|
|
"reduce_lr_patience": 4,
|
|
"reduce_lr_min_lr": 1e-6,
|
|
"early_stopping_enabled": true,
|
|
"early_stopping_monitor": "val_loss",
|
|
"early_stopping_patience": 6,
|
|
"early_stopping_restore_best_weights": true,
|
|
"warmup_enabled": true,
|
|
"warmup_epochs": 5,
|
|
"warmup_start_lr": 1e-6
|
|
}
|