File size: 1,667 Bytes
02c5426 |
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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
train_dataset:
dataset:
name: hr_data_loader
args:
# root_path: /data/kyanchen/datasets/UC/256
# split_file: data_split/UC_split.json
root_path: /data/kyanchen/datasets/AID
split_file: data_split/AID_split.json
split_key: train
cache: none
wrapper:
name: inr_range_scale_sr_warp
args:
max_scale_ratio: 5
patch_size: 48
augment: true
sample_q: 1024
val_mode: false
test_mode: false
return_cell: true
batch_size: 64
num_workers: 4
val_dataset:
dataset:
name: hr_data_loader
args:
# root_path: /data/kyanchen/datasets/UC/256
# split_file: data_split/UC_split.json
root_path: /data/kyanchen/datasets/AID
split_file: data_split/AID_split.json
split_key: val
first_k: 128
cache: none
wrapper:
name: inr_range_scale_sr_warp
args:
max_scale_ratio: 5
patch_size: 48
augment: false
sample_q: 1024
val_mode: true
test_mode: false
return_cell: true
batch_size: 64
num_workers: 4
eval_type: psnr+ssim
data_norm:
inp: {sub: [0.5], div: [0.5]}
gt: {sub: [0.5], div: [0.5]}
model:
name: liif
args:
encoder_spec:
# name: edsr-baseline
# name: rdn
name: rcan
args:
no_upsampling: true
imnet_spec:
name: mlp
args:
out_dim: 3
hidden_list: [256, 256, 256, 256]
optimizer:
name: adamw
args:
lr: 1.e-4
epoch_max: 4000
lr_scheduler:
# name: CosineAnnealingLR
# T_max: 1500
# eta_min: 1.e-7
name: MultiStepLR
milestones: [2500]
gamma: 0.1
epoch_val_interval: 40
epoch_save_interval: 300
|