Upload models/ddpm/level3.0_seed43/config.yaml with huggingface_hub

models/ddpm/level3.0_seed43/config.yaml

@@ -0,0 +1,87 @@
+experiment:
+  name: denoise_ddpm_base0526_level3.0_seed43
+  output_dir: /cloud/cloud-s3fs/ground_roll/results_0526
+  seed: 43
+  device: cuda
+data:
+  segy_pair:
+    input_path: /home/data/noisy/SEGC3_shots1_9_noisy_3.0.sgy
+    target_path: /home/data/noise/SEGC3_shots1_9_noise_3.0.sgy
+    traces_per_shot: 201
+    time_downsample: 1
+  shot_split:
+    train: 7
+    val: 1
+    test: 1
+  loader:
+    batch_size: 110
+    num_workers: 16
+    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: ddpm_unet
+  params:
+    in_channels: 3
+    out_channels: 2
+    base_channels: 64
+    channel_mults:
+    - 1
+    - 2
+    - 4
+    - 8
+    - 8
+    time_emb_dim: 256
+    num_res_blocks: 1
+diffusion:
+  num_timesteps: 1000
+  beta_start: 0.0001
+  beta_end: 0.02
+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: 1
+  eval_interval: 20
+  ckpt_interval: 20
+  vis_interval: 5
+  eval_sample_steps: 20
+  eval_use_ddim: true
+  eval_ddim_eta: 0.0
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5