File size: 1,304 Bytes
32450bd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "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"
  ]
}