/
tsan_config.yml
58 lines (48 loc) · 1.39 KB
/
tsan_config.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
dataset:
test: # should be modified by your own file
type: VideoTestMFQEv2Dataset
gt_path: /home/data/disk1/xuli/Transcode/dataset/test18/raw/raw/eval/
lq_path: /home/data/disk1/xuli/Transcode/dataset/test18/x265/bitrate10/eval/
network:
radius: 3 # total num of input frame = 2 * radius + 1
tsan:
in_nc: 1 # 1 for Y
out_nc: 1
nf: 64 # num of feature maps
nb: 10 # num of conv layers
base_ks: 3
deform_ks: 3 # size of the deformable kernel
train:
exp_name: TSAN # default: timestr. None: ~
random_seed: 7
pre-val: False # evaluate criterion before training, e.g., ori PSNR
num_iter: !!float 4e+5
interval_print: !!float 10
interval_val: !!float 5e+3 # also save model
pbar_len: 100
restore_iter: 0000
checkpoint_save_path: ./exp/TSAN
optim:
type: Adam
lr: !!float 1e-4 # init lr of scheduler
betas: [0.9, 0.999]
eps: !!float 1e-08
scheduler:
is_on: False
type: CosineAnnealingRestartLR
periods: [!!float 5e+4, !!float 5e+4, !!float 5e+4, !!float 5e+4, !!float 5e+4, !!float 5e+4] # epoch interval
restart_weights: [1, 0.5, 0.5, 0.5, 0.5, 0.5]
eta_min: !!float 1e-7
loss:
type: CharbonnierLoss
eps: !!float 1e-6
criterion:
type: PSNR
unit: dB
test:
restore_iter: !!float 300000
pbar_len: 100
save_path: ./results # modify by yourself
criterion:
type: PSNR
unit: dB