| seed: 42 | |
| data: | |
| root: data | |
| format_version: mass_conserving_msw_v1 | |
| grid_points: 250 | |
| domain_km: 125 | |
| variables: [u, h, r] | |
| input_channels: 4 | |
| train_samples: 24 | |
| validation_samples: 12 | |
| model: | |
| input_channels: 4 | |
| hidden_channels: 32 | |
| hidden_layers: 4 | |
| kernel_size: 3 | |
| train: | |
| epochs: 2 | |
| batch_size: 8 | |
| learning_rate: 0.001 | |
| optimizer: Adam | |
| eta: 2.0 | |
| num_workers: 0 | |
| runtime: | |
| device: auto | |
| paths: | |
| checkpoint: result/checkpoints/massconservingcnn.pt | |
| training_metrics: result/training/metrics.json | |
| inference: result/output/predictions.npz | |
| evaluation_dir: result/evaluation | |
| paper_model: | |
| input_channels: 4 | |
| output_channels: 3 | |
| grid_points: 250 | |
| hidden_channels: 32 | |
| hidden_layers: 4 | |
| kernel_size: 3 | |
| hidden_activation: SELU | |
| rain_activation: ReLU | |
| padding: circular | |
| influence_radius: 5 | |
| train_samples: 48000 | |
| validation_samples: 48000 | |
| batch_size: 96 | |
| epochs: 100 | |
| optimizer: Adam | |
| eta: 2.0 | |
| experiment: dT10_eta2 | |