AIFS_Single_v1 / config.json
yzt15806542928's picture
新增config.json文件
32450bd verified
Raw
History Blame Contribute Delete
1.3 kB
{
"model_name": "AIFS Single v1.1",
"model_type": "aifs",
"architectures": [
"AIFS"
],
"framework": "PyTorch with Anemoi",
"domain": "atmosphere",
"task": "global-weather-forecasting",
"implementation": {
"entry_point": "model/aifs.py",
"scope": "AIFS v1.1 checkpoint wrapper with matching from-scratch construction support"
},
"architecture": {
"family": "encoder-processor-decoder graph neural network",
"encoder": "GraphTransformerForwardMapper",
"processor": "TransformerProcessor",
"decoder": "GraphTransformerBackwardMapper",
"hidden_size": 1024,
"processor_layers": 16,
"attention_heads": 16,
"activation": "GELU"
},
"data": {
"grid": "N320 reduced Gaussian",
"grid_points": 542080,
"time_step_hours": 6,
"feature_count": 115,
"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"
]
},
"configuration_sources": [
"model/aifs_config.json",
"model/aifs.py",
"conf/config.yaml"
]
}