File size: 1,195 Bytes
9191802 | 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 | {
"model_name": "FuXi v2.1",
"model_type": "fuxi_v21",
"architectures": [
"FuXi21"
],
"framework": "PyTorch",
"domain": "atmosphere",
"task": "global-medium-range-weather-forecasting",
"implementation": {
"entry_point": "model/FuXi21.py",
"scope": "trainable forward-graph reconstruction with full-resolution and smoke profiles"
},
"architecture": {
"family": "windowed Transformer with rotary attention and PixelShuffle decoder",
"grid_shape": [
721,
1440
],
"input_channels": 85,
"output_channels": 85,
"static_channels": 6,
"patch_size": 6,
"embedding_size": 1536,
"depth": 30,
"attention_heads": 24,
"mlp_size": 4096,
"window_size": 20,
"activation": "SiLU/GELU"
},
"data": {
"dataset": "ERA5",
"spatial_resolution_degrees": 0.25,
"time_step_hours": 6,
"input_steps": 2,
"pressure_levels_hpa": [
50,
100,
150,
200,
250,
300,
400,
500,
600,
700,
850,
925,
1000
],
"protocol": "non_official_protocol"
},
"configuration_sources": [
"conf/config.yaml",
"model/FuXi21.py"
]
}
|