config.json

{
  "model_name": "StormCast",
  "model_type": "stormcast",
  "architectures": [
    "StormCast"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "regional-convection-resolving-weather-forecasting",
  "implementation": {
    "entry_point": "model/stormer.py",
    "scope": "two-stage generative forecasting: a deterministic regression UNet followed by an EDM-preconditioned conditional residual diffusion UNet, both built on the SongUNet backbone"
  },
  "architecture": {
    "family": "two-stage UNet with conditional residual diffusion",
    "stages": [
      "regression",
      "diffusion"
    ],
    "backbone": "SongUNet",
    "image_size": [
      512,
      640
    ],
    "model_channels": 128,
    "channel_mult": [
      1,
      2,
      2,
      2,
      2
    ],
    "num_blocks": 4,
    "attention_resolutions": [],
    "state_channels": 99,
    "background_channels": 26,
    "invariant_channels": 2,
    "regression": {
      "condition": [
        "state",
        "background",
        "invariant"
      ],
      "input_channels": 127,
      "output_channels": 99,
      "additive_pos_embed": false
    },
    "diffusion": {
      "condition": [
        "state",
        "regression",
        "invariant"
      ],
      "input_channels": 299,
      "output_channels": 99,
      "additive_pos_embed": true,
      "preconditioning": "EDM",
      "sigma_data": 0.5,
      "P_mean": -1.2,
      "P_std": 1.2
    }
  },
  "sampling": {
    "diffusion_steps": 18,
    "sigma_min": 0.002,
    "sigma_max": 800.0,
    "rho": 7.0
  },
  "data": {
    "dataset": "ERA5 backgrounds with local high-resolution state",
    "background_source": "ERA5",
    "state_source": "convection-resolving regional reanalysis (HRRR-style)",
    "regional_grid": {
      "projection": "Lambert conformal conic",
      "image_size": [
        512,
        640
      ],
      "horizontal_spacing_km": 3.0,
      "standard_latitude_degrees": 38.5,
      "origin_latitude_degrees": 38.5,
      "central_longitude_degrees": -97.5
    },
    "background_grid": [
      721,
      1440
    ],
    "background_spatial_resolution_degrees": 0.25,
    "time_step_hours": 1,
    "input_steps": 1,
    "output_steps": 1,
    "state_surface_variables": [
      "10m_u_component_of_wind",
      "10m_v_component_of_wind",
      "2m_temperature",
      "mean_sea_level_pressure",
      "composite_reflectivity"
    ],
    "state_model_level_variables": [
      "u_component_of_wind",
      "v_component_of_wind",
      "temperature",
      "specific_humidity",
      "geopotential",
      "pressure"
    ],
    "invariant_variables": [
      "land_sea_mask",
      "orography"
    ]
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/stormer.py",
    "scripts/train.py",
    "scripts/data_loader.py",
    "scripts/grid.py"
  ]
}