File size: 987 Bytes
cc22da3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "model_name": "FourCastNet",
  "model_type": "fourcastnet",
  "architectures": [
    "FourCastNet"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "global-weather-forecasting",
  "implementation": {
    "entry_point": "model/fourcastnet.py",
    "scope": "repository AFNO implementation"
  },
  "architecture": {
    "family": "adaptive Fourier neural operator",
    "model_grid_shape": [
      720,
      1440
    ],
    "patch_size": [
      8,
      8
    ],
    "input_channels": 19,
    "output_channels": 19,
    "embedding_size": 768,
    "depth": 12,
    "mlp_ratio": 4.0,
    "afno_channel_blocks": 8,
    "sparsity_threshold": 0.01,
    "hard_thresholding_fraction": 1.0
  },
  "data": {
    "dataset": "ERA5",
    "nominal_spatial_resolution_degrees": 0.25,
    "time_step_hours": 6,
    "pressure_levels_hpa": [
      50,
      500,
      850,
      1000
    ]
  },
  "configuration_sources": [
    "model/fourcastnet.py",
    "conf/config.yaml"
  ]
}