Upload models/enhanced_atten_unet/level5.0_seed42/config.yaml with huggingface_hub

models/enhanced_atten_unet/level5.0_seed42/config.yaml

@@ -1,12 +1,12 @@
 experiment:
-  name: denoise_enhanced_atten_unet_base0511_level5.0_seed42
-  output_dir: /cloud/cloud-s3fs/ground_roll/results_0510
+  name: denoise_enhanced_atten_unet_base0615_level5.0_seed42
+  output_dir: /home/sharedata/gtx/benchmark/ground_roll/results_0615
   seed: 42
   device: cuda:4
 data:
   segy_pair:
-    input_path: /cloud/cloud-s3fs/data/noisy/SEGC3_shots1_9_noisy_5.0.sgy
-    target_path: /cloud/cloud-s3fs/data/noise/SEGC3_shots1_9_noise_5.0.sgy
+    input_path: /home/sharedata/gtx/benchmark/ground_roll/noisy/SEGC3_shots1_9_noisy_5.0.sgy
+    target_path: /home/sharedata/gtx/benchmark/ground_roll/noise/SEGC3_shots1_9_noise_5.0.sgy
     traces_per_shot: 201
     time_downsample: 1
   shot_split:
@@ -14,7 +14,7 @@ data:
     val: 1
     test: 1
   loader:
-    batch_size: 256
+    batch_size: 80
     num_workers: 4
     pin_memory: true
 preprocess: