Upload models/unet_res/uniform/miss70/seed44/config.yaml with huggingface_hub

models/unet_res/uniform/miss70/seed44/config.yaml

@@ -0,0 +1,86 @@
+experiment:
+  name: interp_unet_res_seed44_uniform_miss70
+  output_dir: /cloud/cloud-s3fs/interpolation
+  seed: 44
+  device: cuda
+data:
+  segy:
+    path: /cloud/cloud-ssd1/SEGC3/SEG_45Shot_shots1-9.sgy
+    traces_per_shot: 201
+    time_downsample: 1
+  test_ratio: 0.1
+  loader:
+    batch_size: 156
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.002
+  t0: 0.0
+  spherical_power: 1.2
+  uniform_stride: 2
+  normalize_mode: max_abs
+  normalize_scope: global
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_shots: null
+  mask_mode: uniform
+  mask_ratio: 0.7
+model:
+  type: res_unet
+  params:
+    in_channels: 1
+    out_channels: 1
+    base_channels: 32
+    depth: 4
+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: mse
+  params: {}
+optim:
+  type: adamw
+  params:
+    lr: 0.001
+    weight_decay: 0.0001
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 20
+  eval_interval: 1
+  ckpt_interval: 5
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  data:
+    segy:
+      path: /cloud/cloud-ssd1/SEGC3/SEG_45Shot_shots10-18_shotper0.3.sgy
+      traces_per_shot: 201
+      time_downsample: 1
+  checkpoint: /cloud/cloud-s3fs/interpolation/interp_unet_base/checkpoints/epoch_200.pt
+  output_dir: /cloud/cloud-s3fs/interpolation/interp_unet_base/inference
+  n_viz_shots: 5
+  device: 1
+  batch_size: 4