File size: 1,727 Bytes
0bac657 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | {
"model_name": "GenCast",
"model_type": "gencast",
"architectures": [
"GenCast"
],
"framework": "JAX with Haiku",
"domain": "atmosphere",
"task": "probabilistic-global-weather-forecasting",
"implementation": {
"entry_point": "model/gencast.py",
"scope": "official-equivalent model API with a configurable mini random-initialization profile"
},
"architecture": {
"family": "graph-based denoising diffusion model",
"mesh_refinement_level": 4,
"mesh_nodes": 2562,
"latent_size": 512,
"mlp_hidden_layers": 1,
"sparse_transformer_layers": 16,
"attention_heads": 4,
"feed_forward_size": 2048,
"attention_k_hop": 16,
"attention_type": "triblockdiag_mha"
},
"data": {
"dataset": "ERA5",
"input_duration_hours": 24,
"forecast_step_hours": 12,
"target_channels": 84,
"pressure_levels_hpa": [
50,
100,
150,
200,
250,
300,
400,
500,
600,
700,
850,
925,
1000
],
"pressure_level_variables": [
"geopotential",
"specific_humidity",
"temperature",
"u_component_of_wind",
"v_component_of_wind",
"vertical_velocity"
],
"surface_target_variables": [
"2m_temperature",
"10m_u_component_of_wind",
"10m_v_component_of_wind",
"mean_sea_level_pressure",
"sea_surface_temperature",
"total_precipitation_12hr"
]
},
"sampling": {
"noise_levels": 20,
"minimum_noise_level": 0.03,
"maximum_noise_level": 80.0,
"rho": 7.0
},
"configuration_sources": [
"conf/config.yaml",
"model/gencast.py",
"model/graphcast/gencast.py",
"model/data_loader.py"
]
}
|