| project: | |
| name: stormcast | |
| seed: 42 | |
| output_dir: ./outputs | |
| data: | |
| root_dir: ./data | |
| train_years: [2019] | |
| val_years: [2021] | |
| test_years: [2022] | |
| time_step_hours: 1 | |
| input_steps: 1 | |
| output_steps: 1 | |
| era5_image_size: [721, 1440] | |
| image_size: [512, 640] | |
| num_timesteps: 4 | |
| normalize: true | |
| era5_variables: | |
| - u10m | |
| - v10m | |
| - t2m | |
| - tcwv | |
| - sp | |
| - msl | |
| - u1000 | |
| - u850 | |
| - u500 | |
| - u250 | |
| - v1000 | |
| - v850 | |
| - v500 | |
| - v250 | |
| - z1000 | |
| - z850 | |
| - z500 | |
| - z250 | |
| - t1000 | |
| - t850 | |
| - t500 | |
| - t250 | |
| - q1000 | |
| - q850 | |
| - q500 | |
| - q250 | |
| state_variables: | |
| - u10m | |
| - v10m | |
| - t2m | |
| - msl | |
| - u1hl | |
| - u2hl | |
| - u3hl | |
| - u4hl | |
| - u5hl | |
| - u6hl | |
| - u7hl | |
| - u8hl | |
| - u9hl | |
| - u10hl | |
| - u11hl | |
| - u13hl | |
| - u15hl | |
| - u20hl | |
| - u25hl | |
| - u30hl | |
| - v1hl | |
| - v2hl | |
| - v3hl | |
| - v4hl | |
| - v5hl | |
| - v6hl | |
| - v7hl | |
| - v8hl | |
| - v9hl | |
| - v10hl | |
| - v11hl | |
| - v13hl | |
| - v15hl | |
| - v20hl | |
| - v25hl | |
| - v30hl | |
| - t1hl | |
| - t2hl | |
| - t3hl | |
| - t4hl | |
| - t5hl | |
| - t6hl | |
| - t7hl | |
| - t8hl | |
| - t9hl | |
| - t10hl | |
| - t11hl | |
| - t13hl | |
| - t15hl | |
| - t20hl | |
| - t25hl | |
| - t30hl | |
| - q1hl | |
| - q2hl | |
| - q3hl | |
| - q4hl | |
| - q5hl | |
| - q6hl | |
| - q7hl | |
| - q8hl | |
| - q9hl | |
| - q10hl | |
| - q11hl | |
| - q13hl | |
| - q15hl | |
| - q20hl | |
| - q25hl | |
| - q30hl | |
| - Z1hl | |
| - Z2hl | |
| - Z3hl | |
| - Z4hl | |
| - Z5hl | |
| - Z6hl | |
| - Z7hl | |
| - Z8hl | |
| - Z9hl | |
| - Z10hl | |
| - Z11hl | |
| - Z13hl | |
| - Z15hl | |
| - Z20hl | |
| - Z25hl | |
| - Z30hl | |
| - p1hl | |
| - p2hl | |
| - p3hl | |
| - p4hl | |
| - p5hl | |
| - p6hl | |
| - p7hl | |
| - p8hl | |
| - p9hl | |
| - p10hl | |
| - p11hl | |
| - p13hl | |
| - p15hl | |
| - p20hl | |
| - refc | |
| invariant_variables: [lsm, orography] | |
| dataloader: | |
| batch_size: 1 | |
| num_workers: 0 | |
| pin_memory: true | |
| model: | |
| regression_conditions: [state, background, invariant] | |
| diffusion_conditions: [state, regression, invariant] | |
| image_size: [512, 640] | |
| model_channels: 128 | |
| channel_mult: [1, 2, 2, 2, 2] | |
| num_blocks: 4 | |
| attention_resolutions: [] | |
| regression_weights: ./data/checkpoint/regression/model_bak.pt | |
| diffusion_weights: ./data/checkpoint/diffusion/model_bak.pt | |
| training: | |
| stage: regression | |
| from_scratch: true | |
| initial_weights: null | |
| resume_checkpoint: null | |
| regression_weights: ./data/checkpoint/regression/model_bak.pt | |
| epochs: 10 | |
| max_steps: 10 | |
| learning_rate: 0.0002 | |
| betas: [0.9, 0.999] | |
| weight_decay: 0.0 | |
| gradient_clip_norm: null | |
| checkpoint_interval: 1 | |
| log_interval: 1 | |
| sigma_data: 0.5 | |
| P_mean: -1.2 | |
| P_std: 1.2 | |
| checkpoint_dir: ./data/checkpoint | |
| inference: | |
| split: test | |
| num_steps: 1 | |
| diffusion_steps: 18 | |
| sigma_min: 0.002 | |
| sigma_max: 800.0 | |
| rho: 7.0 | |
| output_dir: ./outputs/inference | |
| plot_state_variable: refc | |
| plot_background_variable: t2m | |