| experiment: | |
| 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: /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: | |
| train: 7 | |
| val: 1 | |
| test: 1 | |
| loader: | |
| batch_size: 80 | |
| num_workers: 4 | |
| pin_memory: true | |
| preprocess: | |
| normalize_mode: max_abs | |
| normalize_scope: global | |
| patch_time: 256 | |
| patch_trace: 128 | |
| patch_overlap: 0.5 | |
| max_shots: null | |
| model: | |
| type: enhanced_atten_unet | |
| params: | |
| in_channels: 1 | |
| out_channels: 1 | |
| base_channels: 32 | |
| depth: 4 | |
| loss: | |
| type: hybrid_mse_afm | |
| params: | |
| lambda_afm: 0.001 | |
| 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: 200 | |
| grad_clip: 1.0 | |
| log_step: false | |
| log_interval: 10 | |
| eval_interval: 1 | |
| ckpt_interval: 20 | |
| vis_interval: 5 | |
| resume: null | |
| log: | |
| log_dir: logs | |
| plot_interval: 5 | |