models/pix2pix/level5.0_seed44/config.yaml

experiment:
  name: denoise_pix2pix_base0521_level5.0_seed44
  output_dir: /data/shared/benchmark/ground_roll/results_0510
  seed: 44
  device: cuda:4
data:
  segy_pair:
    input_path: /data/shared/benchmark/ground_roll/noisy/SEGC3_shots1_9_noisy_5.0.sgy
    target_path: /data/shared/benchmark/ground_roll/noise/SEGC3_shots1_9_noise_5.0.sgy
    traces_per_shot: 201
    time_downsample: 1
  shot_split:
    train: 7
    val: 1
    test: 1
  loader:
    batch_size: 256
    num_workers: 4
    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: pix2pix_generator
  params:
    in_channels: 1
    out_channels: 1
discriminator:
  in_channels: 1
gan:
  lambda_l1: 100.0
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: adam
  params:
    lr: 0.0002
    betas:
    - 0.5
    - 0.999
d_optim:
  type: adam
  params:
    lr: 0.0002
    betas:
    - 0.5
    - 0.999
scheduler:
  type: cosine
  params:
    min_lr: 1.0e-06
d_scheduler:
  type: cosine
  params:
    min_lr: 1.0e-06
train:
  epochs: 100
  grad_clip: null
  log_step: false
  log_interval: 10
  eval_interval: 1
  ckpt_interval: 20
  vis_interval: 5
  resume: null
log:
  log_dir: logs
  plot_interval: 5