| { | |
| "name": "sysu_cd-train-bifa", | |
| "phase": "train", | |
| "gpu_ids": [ | |
| 0 | |
| ], | |
| "path_cd": { | |
| "log": "logs", | |
| "result": "results", | |
| "checkpoint": "checkpoint", | |
| "resume_state": null | |
| }, | |
| "datasets": { | |
| "train": { | |
| "name": "SYSU-CD", | |
| "datasetroot": "/new-home/buddhiw/CD-Models/generated_dataset_views/sysu_cd/legacy_list", | |
| "resolution": 256, | |
| "num_workers": 4, | |
| "batch_size": 8, | |
| "use_shuffle": true, | |
| "data_len": -1 | |
| }, | |
| "val": { | |
| "name": "SYSU-CD", | |
| "datasetroot": "/new-home/buddhiw/CD-Models/generated_dataset_views/sysu_cd/legacy_list", | |
| "resolution": 256, | |
| "num_workers": 4, | |
| "batch_size": 8, | |
| "use_shuffle": false, | |
| "data_len": -1 | |
| }, | |
| "test": { | |
| "name": "SYSU-CD", | |
| "datasetroot": "/new-home/buddhiw/CD-Models/generated_dataset_views/sysu_cd/legacy_list", | |
| "resolution": 256, | |
| "num_workers": 4, | |
| "batch_size": 8, | |
| "use_shuffle": false, | |
| "data_len": -1 | |
| } | |
| }, | |
| "model": { | |
| "name": "bifa", | |
| "loss": "ce_dice" | |
| }, | |
| "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 | |
| } | |
| } | |
| } |