experiment:
  name: denoise_sanet_base0511_level1.0_seed42
  output_dir: /home/sharedata/gtx/benchmark/ground_roll/results_0510
  seed: 42
  device: cuda:4
data:
  segy_pair:
    input_path: /home/sharedata/gtx/benchmark/ground_roll/noisy/SEGC3_shots1_9_noisy_1.0.sgy
    target_path: /home/sharedata/gtx/benchmark/ground_roll/noise/SEGC3_shots1_9_noise_1.0.sgy
    traces_per_shot: 201
    time_downsample: 1
  shot_split:
    train: 7
    val: 1
    test: 1
  loader:
    batch_size: 64
    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: sanet
  params:
    in_channels: 1
    out_channels: 1
    base_channels: 32
    num_blocks: 8
    kernel_sizes:
    - 3
    - 5
    - 7
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