config.json

{
  "model_name": "Stormer",
  "model_type": "stormer",
  "architectures": [
    "Stormer"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "global-medium-range-weather-forecasting",
  "implementation": {
    "entry_point": "model/stormer.py",
    "scope": "repository interval-conditioned forecasting implementation"
  },
  "architecture": {
    "family": "interval-conditioned vision transformer",
    "input_grid_shape": [
      128,
      256
    ],
    "patch_size": 2,
    "hidden_size": 1024,
    "transformer_layers": 24,
    "attention_heads": 16,
    "mlp_ratio": 4.0,
    "variable_channels": 69,
    "conditioning": "adaLN-Zero forecast interval"
  },
  "data": {
    "dataset": "ERA5",
    "spatial_resolution_degrees": 1.40625,
    "data_frequency_hours": 6,
    "training_intervals_hours": [
      6,
      12,
      24
    ],
    "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"
    ],
    "surface_variable_count": 4
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/stormer.py"
  ]
}