| seed: 42 | |
| data: | |
| directory: data/data | |
| filename: burgers_data_R10.mat | |
| input_key: a | |
| target_key: u | |
| expected_shape: [2048, 8192] | |
| resolution: 256 | |
| n_train: 1600 | |
| n_val: 448 | |
| interpolation: linear | |
| normalization_eps: 1.0e-5 | |
| model: | |
| in_channels: 2 | |
| out_channels: 1 | |
| width: 128 | |
| depth: 8 | |
| scattering: efficient | |
| rate_range: 0.25 | |
| eps: 1.0e-6 | |
| training: | |
| epochs: 500 | |
| batch_size: 4 | |
| learning_rate: 0.001 | |
| weight_decay: 1.0e-5 | |
| betas: [0.9, 0.999] | |
| optimizer_eps: 1.0e-8 | |
| scheduler_step: 5 | |
| scheduler_gamma: 0.96 | |
| grad_clip: 1.0 | |
| num_workers: 0 | |
| log_interval: 50 | |
| device: cuda | |
| checkpoint: weight/best_model.pth | |
| last_checkpoint: weight/last_model.pth | |
| inference: | |
| batch_size: 4 | |
| device: cuda | |
| results: | |
| directory: results | |