File size: 579 Bytes
f2c0c8b
b7e3f31
 
 
f2c0c8b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "data_type": "HG",
  "dataset_name": "MNCD256",
  "dataset_path": "HZDR-FWGEL/MineNetCD256",
  "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/"
  }
}