config.json

{
  "model_name": "Pangu-Weather",
  "model_type": "pangu-weather",
  "architectures": [
    "Pangu"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "global-medium-range-weather-forecasting",
  "implementation": {
    "entry_point": "model/pangu.py",
    "scope": "repository 3D Earth-specific transformer implementation"
  },
  "architecture": {
    "family": "3D Earth-specific transformer",
    "input_grid_shape": [
      721,
      1440
    ],
    "patch_size": [
      2,
      4,
      4
    ],
    "embedding_size": 192,
    "stage_depths": [
      2,
      6,
      6,
      2
    ],
    "stage_attention_heads": [
      6,
      12,
      12,
      6
    ],
    "window_size": [
      2,
      6,
      12
    ],
    "surface_input_channels": 7,
    "predicted_surface_channels": 4,
    "upper_air_variables": 5
  },
  "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",
      "specific_humidity",
      "temperature",
      "u_component_of_wind",
      "v_component_of_wind"
    ]
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/pangu.py"
  ]
}