File size: 1,386 Bytes
2ae1696 | 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 | experiment:
name: denoise_dncnn_base0511_level7.0_seed43
output_dir: /cloud/cloud-s3fs/ground_roll/results_0510
seed: 43
device: cuda
data:
segy_pair:
input_path: /home/data/noisy/SEGC3_shots1_9_noisy_7.0.sgy
target_path: /home/data/noise/SEGC3_shots1_9_noise_7.0.sgy
traces_per_shot: 201
time_downsample: 1
shot_split:
train: 7
val: 1
test: 1
loader:
batch_size: 128
num_workers: 16
pin_memory: true
preprocess:
normalize_mode: max_abs
normalize_scope: global
patch_time: 256
patch_trace: 128
patch_overlap: 0.5
max_shots: null
model:
type: dncnn
params:
in_channels: 1
out_channels: 1
depth: 17
base_channels: 32
kernel_size: 3
loss:
type: mse
params:
reduction: mean
metrics:
- name: snr
params:
reduction: per_sample
- name: psnr
params:
data_range: 1.0
reduction: per_sample
- name: ssim
params:
data_range: 2.0
window_size: 11
sigma: 1.5
- name: mae
params: {}
- name: mse
params: {}
- name: rmse
params:
reduction: per_sample
optim:
type: adamw
params:
lr: 0.0001
weight_decay: 1.0e-05
scheduler:
type: cosine
params:
min_lr: 1.0e-06
train:
epochs: 200
grad_clip: 1.0
log_step: false
log_interval: 10
eval_interval: 1
ckpt_interval: 20
vis_interval: 5
resume: null
log:
log_dir: logs
plot_interval: 5
|