{ "name": "dsifn_cd-train-cdmamba", "phase": "train", "gpu_ids": [ 0 ], "path_cd": { "log": "logs", "result": "results", "checkpoint": "checkpoint", "resume_state": null }, "datasets": { "train": { "name": "DSIFN-CD", "datasetroot": "/new-home/buddhiw/CD-Models/generated_dataset_views/dsifn_cd/legacy_list", "resolution": 256, "num_workers": 2, "batch_size": 8, "use_shuffle": true, "data_len": -1 }, "val": { "name": "DSIFN-CD", "datasetroot": "/new-home/buddhiw/CD-Models/generated_dataset_views/dsifn_cd/legacy_list", "resolution": 256, "num_workers": 2, "batch_size": 8, "use_shuffle": false, "data_len": -1 }, "test": { "name": "DSIFN-CD", "datasetroot": "/new-home/buddhiw/CD-Models/generated_dataset_views/dsifn_cd/legacy_list", "resolution": 256, "num_workers": 2, "batch_size": 8, "use_shuffle": false, "data_len": -1 } }, "model": { "name": "cdmamba", "loss": "ce_dice", "init_filters": 16, "n_classes": 2, "mode": "AGLGF", "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": 50, "val_freq": 1, "val_print_iter": 20, "optimizer": { "type": "adam", "lr": 0.0001 }, "sheduler": { "lr_policy": "linear", "n_step": 3, "gamma": 0.1 } } }