| { |
| "model_name": "FengWu", |
| "model_type": "fengwu", |
| "architectures": [ |
| "Fengwu" |
| ], |
| "framework": "PyTorch", |
| "domain": "atmosphere", |
| "task": "global-medium-range-weather-forecasting", |
| "implementation": { |
| "entry_point": "model/fengwu.py", |
| "scope": "repository multi-modal encoder-fuser-decoder implementation" |
| }, |
| "architecture": { |
| "family": "multi-branch hierarchical transformer", |
| "input_grid_shape": [ |
| 721, |
| 1440 |
| ], |
| "patch_size": [ |
| 4, |
| 4 |
| ], |
| "embedding_size": 192, |
| "stage_attention_heads": [ |
| 6, |
| 12, |
| 12, |
| 6 |
| ], |
| "fuser_window_size": [ |
| 2, |
| 6, |
| 12 |
| ], |
| "variable_branches": [ |
| "surface", |
| "geopotential", |
| "relative_humidity", |
| "temperature", |
| "u_component_of_wind", |
| "v_component_of_wind" |
| ], |
| "input_channels": 189, |
| "output_channels": 189 |
| }, |
| "data": { |
| "dataset": "ERA5", |
| "spatial_resolution_degrees": 0.25, |
| "time_step_hours": 6, |
| "pressure_levels_hpa": [ |
| 1, |
| 2, |
| 3, |
| 5, |
| 7, |
| 10, |
| 20, |
| 30, |
| 50, |
| 70, |
| 100, |
| 125, |
| 150, |
| 175, |
| 200, |
| 225, |
| 250, |
| 300, |
| 350, |
| 400, |
| 450, |
| 500, |
| 550, |
| 600, |
| 650, |
| 700, |
| 750, |
| 775, |
| 800, |
| 825, |
| 850, |
| 875, |
| 900, |
| 925, |
| 950, |
| 975, |
| 1000 |
| ], |
| "pressure_level_variable_count": 5, |
| "surface_variable_count": 4 |
| }, |
| "configuration_sources": [ |
| "conf/config.yaml", |
| "model/fengwu.py" |
| ] |
| } |
|
|
