experiment:
  name: denoise_physics_base0526_level9.0_seed44
  output_dir: /cloud/cloud-s3fs/ground_roll/results_0526
  seed: 44
  device: cuda
data:
  segy_pair:
    input_path: /home/data/noisy/SEGC3_shots1_9_noisy_9.0.sgy
    target_path: /home/data/noise/SEGC3_shots1_9_noise_9.0.sgy
    traces_per_shot: 201
    time_downsample: 1
  shot_split:
    train: 7
    val: 1
    test: 1
  loader:
    batch_size: 128
    num_workers: 4
    pin_memory: true
preprocess:
  normalize_mode: max_abs
  normalize_scope: shot
  patch_time: 256
  patch_trace: 128
  patch_overlap: 0.5
  max_shots: null
model:
  type: physics_unet
  params:
    in_channels: 1
    out_channels: 1
    base_channels: 32
    num_levels: 3
fk_classifier:
  base_channels: 32
  dropout: 0.5
  lr: 0.0001
  epochs: 50
  batch_size: 64
physics_loss:
  lambda_signal_class: 0.1
  lambda_noise_class: 0.1
  lambda_recover_data: 0.5
  lambda_recover_noise_class: 0.1
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.0001
scheduler:
  type: cosine
  params:
    min_lr: 1.0e-06
train:
  epochs: 100
  grad_clip: 1.0
  log_step: false
  log_interval: 10
  eval_interval: 1
  ckpt_interval: 20
  vis_interval: 20
  resume: null
log:
  log_dir: logs
  plot_interval: 5