| 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 | |