| { | |
| "data_type": "HG", | |
| "dataset_name": "CLCD256", | |
| "dataset_path": "ericyu/CLCD_Cropped_256", | |
| "device": "cuda", | |
| "eval": { | |
| "batch_size": 32 | |
| }, | |
| "input_nc": 3, | |
| "model_name": "ICIFNet", | |
| "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/" | |
| } | |
| } |