-
Notifications
You must be signed in to change notification settings - Fork 0
/
option.yml
51 lines (46 loc) · 1.03 KB
/
option.yml
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
name: Network
model: CF2N
nEpochs: 600
save_best: True
is_cuda: True
gpus: [0, 1]
num_worker: 0
batch_size: 32
hyperparameter:
lambda1: 0.2
lambda2: 0.3
lambda3: 0.5
lambda4: 0.01
seed: 42
weight: weight
checkpoint: checkpoints
training_set: ./datasets/wv3/train
satellite: wv3 # [gf2, qb, wv3]
satellite4train: train_wv3.h5 # [gf2, qb, wv3]
satellite4valid: valid_wv3.h5
pretrain:
pretrained: False
pre_folder: checkpoints
pre_source: 585.pth
test:
model: CF2N
test_set: ./datasets/wv3/test
test4refer: test_wv3_multiExm1.h5
test4nonrefer: test_wv3_OrigScale_multiExm1.h5
type: rr # [rr, fr]
use_metric: True
result2img: False
img_type: mat # [mat, tif]
weight: wv3.pth
schedule:
lr: 0.0003
decay: '100'
gamma: 0.9
optimizer: ADAM # [ADAM, ADAMW, SGD, RMSprop]
momentum: 0.9 # SGD momemtum
beta1: 0.9 # ADAM beta
beta2: 0.999 # ADAM beta
epsilon: 1e-8 # ADAM epsilon
alpha: 0.9 # RMSprop alpha
weight_decay: 0
loss: L1 # [MSE, L1]