config.json

{
  "model_name": "FuXi",
  "model_type": "fuxi",
  "architectures": [
    "Fuxi"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "global-medium-range-weather-forecasting",
  "implementation": {
    "entry_point": "model/fuxi.py",
    "scope": "repository short-, medium-, and long-stage implementation"
  },
  "architecture": {
    "family": "cube embedding with U-Transformer",
    "input_shape": [
      2,
      721,
      1440
    ],
    "input_shape_order": [
      "time",
      "latitude",
      "longitude"
    ],
    "patch_size": [
      2,
      4,
      4
    ],
    "input_channels": 70,
    "output_channels": 70,
    "embedding_size": 1536,
    "normalization_groups": 32,
    "attention_heads": 8,
    "window_size": 7
  },
  "data": {
    "dataset": "ERA5",
    "spatial_resolution_degrees": 0.25,
    "time_step_hours": 6,
    "pressure_levels_hpa": [
      50,
      100,
      150,
      200,
      250,
      300,
      400,
      500,
      600,
      700,
      850,
      925,
      1000
    ],
    "pressure_level_variables": [
      "geopotential",
      "relative_humidity",
      "temperature",
      "u_component_of_wind",
      "v_component_of_wind"
    ],
    "surface_variable_count": 5
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/fuxi.py"
  ]
}