File size: 1,327 Bytes
f57fb08 | 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 | {
"model_name": "ClimaX",
"model_type": "climax",
"architectures": [
"ClimaX"
],
"framework": "PyTorch",
"domain": "climate-and-atmosphere",
"task": "global-weather-forecasting",
"implementation": {
"entry_point": "model/ClimaX.py",
"scope": "repository forecasting configuration"
},
"architecture": {
"family": "variable-tokenized vision transformer",
"input_grid_shape": [
32,
64
],
"patch_size": 2,
"embedding_size": 1024,
"transformer_layers": 8,
"decoder_layers": 2,
"attention_heads": 16,
"mlp_ratio": 4.0,
"input_channels": 48
},
"data": {
"dataset": "ERA5",
"spatial_resolution_degrees": 5.625,
"time_step_hours": 6,
"default_forecast_lead_hours": 6,
"pressure_levels_hpa": [
50,
250,
500,
600,
700,
850,
925
],
"pressure_level_variables": [
"geopotential",
"relative_humidity",
"specific_humidity",
"temperature",
"u_component_of_wind",
"v_component_of_wind"
],
"output_variables": [
"geopotential_500",
"temperature_850",
"2m_temperature",
"10m_u_component_of_wind",
"10m_v_component_of_wind"
]
},
"configuration_sources": [
"conf/config.yaml",
"model/ClimaX.py"
]
}
|