File size: 1,609 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 87 88 89 90 91 92 93 94 95 |
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: cnn_fixed_scale_sr_warp
args:
scale_ratio: 2
patch_size: 48
augment: true
val_mode: false
test_mode: false
batch_size: 96
num_workers: 8
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
cache: none
wrapper:
name: cnn_fixed_scale_sr_warp
args:
scale_ratio: 2
patch_size: 48
augment: false
val_mode: true
test_mode: false
batch_size: 32
num_workers: 4
data_norm:
img: {sub: [0.5], div: [0.5]}
gt: {sub: [0.5], div: [0.5]}
#model:
# name: TransENet
# args:
# scale_ratio: 2
#model:
# name: SRCNN
# args:
# scale_ratio: 2
#model:
# name: FSRCNN
# args:
# scale_ratio: 2
#model:
# name: LGCNET
# args:
# scale_ratio: 2
#model:
# name: DCM
# args:
# scale_ratio: 2
model:
name: VDSR
args:
scale_ratio: 2
optimizer:
name: adamw
args:
lr: 0.0001
epoch_max: 1300
#epoch_max: 2000
lr_scheduler:
# name: CosineAnnealingLR
# T_max: 300
# eta_min: 1.e-6
name: MultiStepLR
milestones: [1000]
gamma: 0.5
epoch_val_interval: 50
epoch_save_interval: 300
|