|
train_dataset: |
|
dataset: |
|
name: hr_data_loader |
|
args: |
|
|
|
|
|
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: 48 |
|
num_workers: 4 |
|
|
|
val_dataset: |
|
dataset: |
|
name: hr_data_loader |
|
args: |
|
|
|
|
|
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: 48 |
|
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: aliif |
|
args: |
|
encoder_spec: |
|
|
|
|
|
name: rcan |
|
args: |
|
no_upsampling: true |
|
imnet_spec: |
|
name: mlp_pw |
|
args: |
|
out_dim: 3 |
|
hidden_list: [16, 16, 16, 16] |
|
pdn_spec: |
|
name: ExpansionNet |
|
args: |
|
in_dim: 580 |
|
out_dim: 10 |
|
hidden_list: [ 256, 256, 256, 256 ] |
|
basis_spec: |
|
name: basis |
|
args: |
|
basis_num: 10 |
|
hidden: 16 |
|
state: train |
|
path: ./save/test/epoch-last.pth |
|
|
|
optimizer: |
|
name: adamw |
|
args: |
|
lr: 1.e-4 |
|
|
|
epoch_max: 3000 |
|
|
|
lr_scheduler: |
|
|
|
|
|
|
|
name: MultiStepLR |
|
milestones: [2000] |
|
gamma: 0.1 |
|
|
|
epoch_val_interval: 40 |
|
epoch_save_interval: 300 |
|
|
