| { |
| "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" |
| ] |
| } |
|
|
