| seed: 42 | |
| data: | |
| path: data/corrdiff.npz | |
| protocol: corrdiff_npz_v1 | |
| input_key: input | |
| target_key: target | |
| input_shape: [12, 36, 36] | |
| target_shape: [4, 448, 448] | |
| input_variables: [tcwv, t2m, u10m, v10m, t500, z500, u500, v500, t850, z850, u850, v850] | |
| target_variables: [t2m, u10m, v10m, maximum_radar_reflectivity] | |
| fake_samples: 2 | |
| model: | |
| in_channels: 12 | |
| out_channels: 4 | |
| base_channels: 8 | |
| feature_size: 28 | |
| output_size: 448 | |
| sigma_data: 0.5 | |
| training: | |
| regression_epochs: 1 | |
| diffusion_epochs: 1 | |
| batch_size: 1 | |
| max_batches_per_epoch: 1 | |
| learning_rate: 0.0002 | |
| num_workers: 0 | |
| amp: true | |
| p_mean: -1.2 | |
| p_std: 1.2 | |
| sampling: | |
| batch_size: 1 | |
| ensemble_size: 2 | |
| steps: 3 | |
| sigma_min: 0.002 | |
| sigma_max: 5.0 | |
| rho: 7.0 | |
| solver: heun | |
| runtime: | |
| device: auto | |
| paths: | |
| checkpoint: result/checkpoints/corrdiff.pt | |
| training_metrics: result/training/metrics.json | |
| predictions: result/output/predictions.npz | |
| evaluation_dir: result/evaluation | |