| { |
| "data_type": "HG", |
| "dataset_name": "MNCD256", |
| "dataset_path": "HZDR-FWGEL/MineNetCD256", |
| "device": "cuda", |
| "eval": { |
| "batch_size": 32 |
| }, |
| "input_nc": 3, |
| "model_name": "RDPNet", |
| "num_classes": 2, |
| "push_to_hub": false, |
| "test": { |
| "batch_size": 32, |
| "checkpoint": "checkpoints/GVLM256/DMINet/\"epoch 99/pytorch_model.bin", |
| "save_path": "./results/" |
| }, |
| "train": { |
| "batch_size": 32, |
| "epochs": 100, |
| "loss": "CrossEntropy", |
| "lr": 0.0001, |
| "optim": "Adam", |
| "save_intervals": 10, |
| "snapshots_dir": "./checkpoints/" |
| } |
| } |
