BasicSR
1# general settings
2name: 103_EDVR_L_x4_SR_REDS_woTSA_600k_B4G8_valREDS4_wandb
3model_type: EDVRModel
4scale: 4
5num_gpu: 8 # set num_gpu: 0 for cpu mode
6manual_seed: 10
7
8# dataset and data loader settings
9datasets:
10train:
11name: REDS
12type: REDSDataset
13dataroot_gt: datasets/REDS/train_sharp
14dataroot_lq: datasets/REDS/train_sharp_bicubic
15dataroot_flow: ~
16meta_info_file: basicsr/data/meta_info/meta_info_REDS_GT.txt
17val_partition: REDS4 # set to 'official' when use the official validation partition
18io_backend:
19type: disk
20
21num_frame: 5
22gt_size: 256
23interval_list: [1]
24random_reverse: false
25use_hflip: true
26use_rot: true
27
28# data loader
29num_worker_per_gpu: 3
30batch_size_per_gpu: 4
31dataset_enlarge_ratio: 200
32prefetch_mode: ~
33
34val:
35name: REDS4
36type: VideoTestDataset
37dataroot_gt: datasets/REDS/train_sharp
38dataroot_lq: datasets/REDS/train_sharp_bicubic
39meta_info_file: basicsr/data/meta_info/meta_info_REDS4_test_GT.txt
40# change to 'meta_info_REDSofficial4_test_GT' when use the official validation partition
41io_backend:
42type: disk
43
44cache_data: false
45num_frame: 5
46padding: reflection_circle
47
48# network structures
49network_g:
50ema_decay: 0.999
51type: EDVR
52num_in_ch: 3
53num_out_ch: 3
54num_feat: 128
55num_frame: 5
56deformable_groups: 8
57num_extract_block: 5
58num_reconstruct_block: 40
59center_frame_idx: ~
60hr_in: false
61with_predeblur: false
62with_tsa: false
63
64# path
65path:
66pretrain_network_g: ~
67strict_load_g: true
68resume_state: ~
69
70# training settings
71train:
72optim_g:
73type: Adam
74lr: !!float 4e-4
75weight_decay: 0
76betas: [0.9, 0.99]
77
78scheduler:
79type: CosineAnnealingRestartLR
80periods: [150000, 150000, 150000, 150000]
81restart_weights: [1, 0.5, 0.5, 0.5]
82eta_min: !!float 1e-7
83
84total_iter: 600000
85warmup_iter: -1 # no warm up
86dcn_lr_mul: 1
87
88# losses
89pixel_opt:
90type: CharbonnierLoss
91loss_weight: 1.0
92reduction: sum
93
94# validation settings
95val:
96val_freq: !!float 5e3
97save_img: false
98
99metrics:
100psnr: # metric name, can be arbitrary
101type: calculate_psnr
102crop_border: 0
103test_y_channel: false
104
105# logging settings
106logger:
107print_freq: 100
108save_checkpoint_freq: !!float 5e3
109use_tb_logger: true
110wandb:
111project: ~
112resume_id: ~
113
114# dist training settings
115dist_params:
116backend: nccl
117port: 29500
118