-
Notifications
You must be signed in to change notification settings - Fork 1
/
option.yml
59 lines (52 loc) · 1.16 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
52
53
54
55
56
57
58
#
# @Author: wjm
# @Date: 2019-10-13 21:45:10
# @LastEditTime: 2019-10-13 21:46:06
# @Description: file content
#
name: Net
algorithm: HMB
nEpochs: 2000
gpu_mode: False
save_best: False
gpus: [0, 1]
threads: 8
seed: 123
checkpoint: checkpoint
data_dir_train: ./dataset/train
data_dir_eval: ./dataset/test
source_ms: ms
source_pan: pan
pretrain:
pretrained: False
pre_sr: 1592134968/best.pth
pre_folder: checkpoint
test:
algorithm: HMB
type: test #[test, eval]
data_dir: ./dataset/test
source_ms: ms
source_pan: pan
model: 1605255794/best.pth
data:
upsacle: 4
batch_size: 1
patch_size: 24
data_augmentation: False
n_colors: 4
rgb_range: 255
normalize : False
schedule:
lr: 1e-4
decay: 1000
gamma: 0.1
optimizer: ADAM #[ADAM, SGD, RMSprop]
momentum: 0.9 #SGD momentum
beta1: 0.9 #ADAM beta
beta2: 0.999 #ADAM beta
epsilon: 1e-8 #ADAM epsilon
alpha: 0.9 #RMSprop alpha
weight_dency: 0
gclip: 0.04
loss: L1 #[MSE, L1]
use_YCbCr: True