File size: 1,763 Bytes
703fb84 |
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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
{
"name": "SYSU-test",
"phase": "test", //train or test
"gpu_ids":[0],
"path_cd": {//set cd path
"log": "logs",
"result": "results",
"checkpoint": "checkpoint",
//"resume_state": null
"resume_state": "experiments/sysu_checkpoint/best_cd_model"
},
"datasets": {
"train": {
"name": "SYSU-CD-256",
"datasetroot": "",
"resolution": 256,
"num_workers": 2,
"batch_size": 6,
"use_shuffle": true,
"data_len": -1
},
"val": {
"name": "SYSU-CD-256",
"datasetroot": "",
"resolution": 256,
"num_workers": 2,
"batch_size": 6,
"use_shuffle": true,
"data_len": -1
},
"test": {
"name": "SYSU-CD-256",
"datasetroot": "",
"resolution": 256,
"num_workers": 2,
"batch_size": 6,
"use_shuffle": false,
"data_len": -1
}
},
"model" : {
"name" : "stnr",
"loss" : "ce_dice",
"init_filters" : 16,
"n_classes" : 2,
"mode" : "FUSION",
"conv_mode" : "orignal_dinner",
"local_query_model" : "orignal_dinner",
"up_mode" : "SRCM",
"up_conv_mode" : "deepwise",
"spatial_dims" : 2,
"in_channels" : 3,
"resdiual" : false,
"blocks_down" : [1, 2, 2, 4],
"blocks_up" : [1, 1, 1],
"diff_abs" : "later",
"stage" : 2,
"mamba_act" : "relu",
"norm" : ["GROUP", {"num_groups": 8}]
},
"train" : {
"n_epoch" : 200,
"train_print_iter" : 500,
"val_freq" : 1,
"val_print_iter" : 50,
"optimizer" :{
"type" : "adam",
"lr" : 1e-4
},
"scheduler" : {
"lr_policy" : "linear",
"n_step" : 3,
"gamma" : 0.1
}
}
} |