| { |
| "era5_variables": [ |
| "u10m", |
| "v10m", |
| "u100m", |
| "v100m", |
| "t2m", |
| "sp", |
| "msl", |
| "u300", |
| "u400", |
| "u500", |
| "u600", |
| "u700", |
| "u850", |
| "u925", |
| "u1000", |
| "v300", |
| "v400", |
| "v500", |
| "v600", |
| "v700", |
| "v850", |
| "v925", |
| "v1000", |
| "z300", |
| "z400", |
| "z500", |
| "z600", |
| "z700", |
| "z850", |
| "z925", |
| "z1000", |
| "t300", |
| "t400", |
| "t500", |
| "t600", |
| "t700", |
| "t850", |
| "t925", |
| "t1000", |
| "q300", |
| "q400", |
| "q500", |
| "q600", |
| "q700", |
| "q850", |
| "q925", |
| "q1000" |
| ], |
| "output_variables": [ |
| "10U", |
| "10V", |
| "2MT", |
| "TD_2M", |
| "TOT_PRECIP", |
| "CLCT", |
| "ASWDIFD_S", |
| "ASWDIR_S", |
| "PS", |
| "H_PBL", |
| "U3D_L45", |
| "U3D_L46", |
| "U3D_L47", |
| "U3D_L48", |
| "U3D_L49", |
| "U3D_L50", |
| "V3D_L45", |
| "V3D_L46", |
| "V3D_L47", |
| "V3D_L48", |
| "V3D_L49", |
| "V3D_L50" |
| ], |
| "pre_invariant_variables": [ |
| "sin_lat", |
| "cos_lat", |
| "sin_lon", |
| "cos_lon", |
| "elevation_norm", |
| "land_fraction" |
| ], |
| "post_invariant_variables": [ |
| "z0_lu_norm", |
| "slope_east_norm", |
| "slope_north_norm" |
| ], |
| "channel_transforms": { |
| "TOT_PRECIP": { |
| "transform": "log_eps", |
| "eps": 1e-05, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=1e-05" |
| }, |
| "ASWDIR_S": { |
| "transform": "log_eps", |
| "eps": 1.0, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=1.0" |
| }, |
| "ASWDIFD_S": { |
| "transform": "log_eps", |
| "eps": 1.0, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=1.0" |
| }, |
| "CLCT": { |
| "transform": "logit_eps_percent", |
| "eps": 0.01, |
| "scale": 100.0, |
| "forward": "logit(eps + (1 - 2*eps) * (x/100))", |
| "inverse": "((sigmoid(y) - eps) / (1 - 2*eps)) * 100, clip to [0, 100]" |
| }, |
| "H_PBL": { |
| "transform": "log_eps", |
| "eps": 10.0, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=10.0" |
| } |
| }, |
| "modes": { |
| "mean": { |
| "stats": "output_physical", |
| "channel_transforms": false, |
| "space": "physical" |
| }, |
| "diffusion": { |
| "stats": "output", |
| "channel_transforms": true, |
| "space": "transformed" |
| } |
| }, |
| "constraints": { |
| "bounds": { |
| "TOT_PRECIP": { |
| "min": 0.0, |
| "max": null, |
| "units": "mm/h" |
| }, |
| "ASWDIFD_S": { |
| "min": 0.0, |
| "max": null, |
| "units": "W/m^2" |
| }, |
| "ASWDIR_S": { |
| "min": 0.0, |
| "max": null, |
| "units": "W/m^2" |
| }, |
| "CLCT": { |
| "min": 0.0, |
| "max": 100.0, |
| "units": "%", |
| "mode": "clamp" |
| }, |
| "PS": { |
| "min": 0.0, |
| "max": null, |
| "units": "Pa" |
| }, |
| "2MT": { |
| "min": 0.0, |
| "max": null, |
| "units": "K" |
| }, |
| "TD_2M": { |
| "min": 0.0, |
| "max": null, |
| "units": "K" |
| }, |
| "H_PBL": { |
| "min": 0.0, |
| "max": null, |
| "units": "m" |
| } |
| }, |
| "sza_gate": { |
| "half_width": 0.05, |
| "channels": { |
| "ASWDIFD_S": { |
| "threshold": 0.01, |
| "half_width": 0.02 |
| }, |
| "ASWDIR_S": { |
| "threshold": 0.05 |
| } |
| } |
| } |
| }, |
| "invariants": { |
| "file": "invariants_rea2_ext.nc", |
| "norm_stats_file": "invariants_norm_stats.json", |
| "channels": [ |
| "elevation", |
| "land_fraction", |
| "z0_lu", |
| "slope_east", |
| "slope_north" |
| ], |
| "native_offset": [ |
| 1000, |
| 1000 |
| ], |
| "native_shape": [ |
| 780, |
| 724 |
| ], |
| "extended_shape": [ |
| 2780, |
| 2724 |
| ], |
| "grid": { |
| "rotated_pole_lat": 40.0, |
| "rotated_pole_lon": -170.0, |
| "dlat": 0.018, |
| "dlon": 0.018, |
| "startlat_ext": -24.0, |
| "startlon_ext": -25.5 |
| }, |
| "values": "physical; wrapper z-scores with norm_stats_file after sampling" |
| }, |
| "wind_levels": { |
| "elevation_invariant": "elevation_norm", |
| "levels": [ |
| { |
| "u": "U3D_L45", |
| "v": "V3D_L45", |
| "a": 252.2799, |
| "b": -9.6701 |
| }, |
| { |
| "u": "U3D_L46", |
| "v": "V3D_L46", |
| "a": 179.7021, |
| "b": -6.8879 |
| }, |
| { |
| "u": "U3D_L47", |
| "v": "V3D_L47", |
| "a": 119.5022, |
| "b": -4.5769 |
| }, |
| { |
| "u": "U3D_L48", |
| "v": "V3D_L48", |
| "a": 71.3555, |
| "b": -2.7345 |
| }, |
| { |
| "u": "U3D_L49", |
| "v": "V3D_L49", |
| "a": 34.8919, |
| "b": -1.3361 |
| }, |
| { |
| "u": "U3D_L50", |
| "v": "V3D_L50", |
| "a": 9.7612, |
| "b": -0.369 |
| } |
| ] |
| }, |
| "number_of_samples": 1, |
| "checkpoints": { |
| "rea2_mean": "regression.mdlus", |
| "rea2_diffusion": "diffusion.mdlus" |
| }, |
| "regression": { |
| "hidden_size": 576, |
| "depth": 16, |
| "num_heads": 9, |
| "patch_size": 2, |
| "attn_kernel_size": 23, |
| "rope_theta": 10000.0, |
| "img_resolution": [ |
| 256, |
| 256 |
| ], |
| "qk_norm": false, |
| "layernorm_backend": "torch", |
| "detok_conv_layers": 2 |
| }, |
| "diffusion": { |
| "hidden_size": 768, |
| "depth": 16, |
| "num_heads": 12, |
| "patch_size": 2, |
| "attn_kernel_size": 23, |
| "sigma_data": 1.0, |
| "rope_theta": 10000.0, |
| "img_resolution": [ |
| 256, |
| 256 |
| ], |
| "qk_norm": false, |
| "layernorm_backend": "torch" |
| }, |
| "sampler": { |
| "num_steps": 18, |
| "sigma_min": 0.002, |
| "sigma_max": 800.0, |
| "rho": 7.0, |
| "solver": "heun" |
| } |
| } |
|
|
