| { |
| "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": [ |
| "U_10M", |
| "V_10M", |
| "T_2M", |
| "PS", |
| "TOT_PRECIP", |
| "CLCT", |
| "ASWDIFD_S", |
| "ASWDIR_S", |
| "ALWU_S", |
| "ATHD_S", |
| "QV_2M", |
| "VMAX_10M", |
| "LHFL_S", |
| "SHFL_S", |
| "U_L35", |
| "U_L36", |
| "U_L37", |
| "U_L38", |
| "U_L39", |
| "U_L40", |
| "V_L35", |
| "V_L36", |
| "V_L37", |
| "V_L38", |
| "V_L39", |
| "V_L40", |
| "T_L35", |
| "T_L36", |
| "T_L37", |
| "T_L38", |
| "T_L39", |
| "T_L40", |
| "TKE_L35", |
| "TKE_L36", |
| "TKE_L37", |
| "TKE_L38", |
| "TKE_L39", |
| "TKE_L40", |
| "Q_L35", |
| "Q_L36", |
| "Q_L37", |
| "Q_L38", |
| "Q_L39", |
| "Q_L40", |
| "H_PBL" |
| ], |
| "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", |
| "cont_50km", |
| "dist_coast_km_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" |
| }, |
| "VMAX_10M": { |
| "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]" |
| }, |
| "LHFL_S": { |
| "transform": "asinh", |
| "eps": 20.0, |
| "forward": "arcsinh(x / eps)", |
| "inverse": "eps * sinh(y) with eps=20.0" |
| }, |
| "SHFL_S": { |
| "transform": "asinh", |
| "eps": 25.0, |
| "forward": "arcsinh(x / eps)", |
| "inverse": "eps * sinh(y) with eps=25.0" |
| }, |
| "TKE_L35": { |
| "transform": "log_eps", |
| "eps": 0.0001, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=0.0001" |
| }, |
| "TKE_L36": { |
| "transform": "log_eps", |
| "eps": 0.0001, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=0.0001" |
| }, |
| "TKE_L37": { |
| "transform": "log_eps", |
| "eps": 0.0001, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=0.0001" |
| }, |
| "TKE_L38": { |
| "transform": "log_eps", |
| "eps": 0.0001, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=0.0001" |
| }, |
| "TKE_L39": { |
| "transform": "log_eps", |
| "eps": 0.0001, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=0.0001" |
| }, |
| "TKE_L40": { |
| "transform": "log_eps", |
| "eps": 0.0001, |
| "forward": "log(x / eps + 1)", |
| "inverse": "eps * (exp(y) - 1) with eps=0.0001" |
| }, |
| "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": { |
| "T_2M": { |
| "min": 0.0, |
| "max": null, |
| "units": "K" |
| }, |
| "PS": { |
| "min": 0.0, |
| "max": null, |
| "units": "Pa" |
| }, |
| "TOT_PRECIP": { |
| "min": 0.0, |
| "max": null, |
| "units": "mm/h" |
| }, |
| "QV_2M": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "VMAX_10M": { |
| "min": 0.0, |
| "max": null, |
| "units": "m/s" |
| }, |
| "TKE_L35": { |
| "min": 0.0, |
| "max": null, |
| "units": "m^2/s^2" |
| }, |
| "TKE_L36": { |
| "min": 0.0, |
| "max": null, |
| "units": "m^2/s^2" |
| }, |
| "TKE_L37": { |
| "min": 0.0, |
| "max": null, |
| "units": "m^2/s^2" |
| }, |
| "TKE_L38": { |
| "min": 0.0, |
| "max": null, |
| "units": "m^2/s^2" |
| }, |
| "TKE_L39": { |
| "min": 0.0, |
| "max": null, |
| "units": "m^2/s^2" |
| }, |
| "TKE_L40": { |
| "min": 0.0, |
| "max": null, |
| "units": "m^2/s^2" |
| }, |
| "Q_L35": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "Q_L36": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "Q_L37": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "Q_L38": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "Q_L39": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "Q_L40": { |
| "min": 0.0, |
| "max": null, |
| "units": "kg/kg" |
| }, |
| "H_PBL": { |
| "min": 0.0, |
| "max": null, |
| "units": "m" |
| }, |
| "CLCT": { |
| "min": 0.0, |
| "max": 100.0, |
| "units": "%", |
| "mode": "clamp" |
| }, |
| "ASWDIFD_S": { |
| "min": 0.0, |
| "max": null, |
| "units": "W/m^2" |
| }, |
| "ASWDIR_S": { |
| "min": 0.0, |
| "max": null, |
| "units": "W/m^2" |
| }, |
| "ALWU_S": { |
| "min": 0.0, |
| "max": null, |
| "units": "W/m^2" |
| }, |
| "ATHD_S": { |
| "min": 0.0, |
| "max": null, |
| "units": "W/m^2" |
| } |
| }, |
| "sza_gate": { |
| "half_width": 0.05, |
| "channels": { |
| "ASWDIFD_S": { |
| "threshold": 0.01, |
| "half_width": 0.02 |
| }, |
| "ASWDIR_S": { |
| "threshold": 0.05 |
| } |
| } |
| } |
| }, |
| "invariants": { |
| "file": "invariants_rea6_ext.nc", |
| "norm_stats_file": "invariants_norm_stats.json", |
| "channels": [ |
| "elevation", |
| "land_fraction", |
| "z0_lu", |
| "slope_east", |
| "slope_north", |
| "cont_50km", |
| "dist_coast_km" |
| ], |
| "native_offset": [ |
| 91, |
| 91 |
| ], |
| "native_shape": [ |
| 824, |
| 848 |
| ], |
| "extended_shape": [ |
| 1006, |
| 1030 |
| ], |
| "grid": { |
| "rotated_pole_lat": 39.25, |
| "rotated_pole_lon": -162.0, |
| "dlat": 0.055, |
| "dlon": 0.055, |
| "startlat_ext": -28.407999999999998, |
| "startlon_ext": -33.408 |
| }, |
| "values": "physical; wrapper z-scores with norm_stats_file after sampling" |
| }, |
| "wind_levels": { |
| "elevation_invariant": "elevation_norm", |
| "levels": [ |
| { |
| "u": "U_L35", |
| "v": "V_L35", |
| "a": 253.9043, |
| "b": -7.7747 |
| }, |
| { |
| "u": "U_L36", |
| "v": "V_L36", |
| "a": 175.3435, |
| "b": -5.3784 |
| }, |
| { |
| "u": "U_L37", |
| "v": "V_L37", |
| "a": 113.9468, |
| "b": -3.494 |
| }, |
| { |
| "u": "U_L38", |
| "v": "V_L38", |
| "a": 67.7674, |
| "b": -2.0716 |
| }, |
| { |
| "u": "U_L39", |
| "v": "V_L39", |
| "a": 33.8841, |
| "b": -1.0358 |
| }, |
| { |
| "u": "U_L40", |
| "v": "V_L40", |
| "a": 9.8204, |
| "b": -0.2995 |
| } |
| ] |
| }, |
| "number_of_samples": 1, |
| "checkpoints": { |
| "rea6_mean": "regression.mdlus", |
| "rea6_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": 1600.0, |
| "rho": 7.0, |
| "solver": "heun" |
| } |
| } |
|
|
