{
  "model_name": "Sonny",
  "sonny_version": "1.0",
  "architecture_name": "Sonny",
  "architecture_family": "StepNets",
  "class_qualname": "networks.sonny.Sonny",
  "source_checkpoint": "stormbreaker5_checkpoint_ema_epoch_050.pth",
  "source_epoch": 50,
  "init_args": {
    "variables": [
      "2m_temperature",
      "10m_u_component_of_wind",
      "10m_v_component_of_wind",
      "mean_sea_level_pressure",
      "geopotential_50",
      "geopotential_100",
      "geopotential_150",
      "geopotential_200",
      "geopotential_250",
      "geopotential_300",
      "geopotential_400",
      "geopotential_500",
      "geopotential_600",
      "geopotential_700",
      "geopotential_850",
      "geopotential_925",
      "geopotential_1000",
      "u_component_of_wind_50",
      "u_component_of_wind_100",
      "u_component_of_wind_150",
      "u_component_of_wind_200",
      "u_component_of_wind_250",
      "u_component_of_wind_300",
      "u_component_of_wind_400",
      "u_component_of_wind_500",
      "u_component_of_wind_600",
      "u_component_of_wind_700",
      "u_component_of_wind_850",
      "u_component_of_wind_925",
      "u_component_of_wind_1000",
      "v_component_of_wind_50",
      "v_component_of_wind_100",
      "v_component_of_wind_150",
      "v_component_of_wind_200",
      "v_component_of_wind_250",
      "v_component_of_wind_300",
      "v_component_of_wind_400",
      "v_component_of_wind_500",
      "v_component_of_wind_600",
      "v_component_of_wind_700",
      "v_component_of_wind_850",
      "v_component_of_wind_925",
      "v_component_of_wind_1000",
      "temperature_50",
      "temperature_100",
      "temperature_150",
      "temperature_200",
      "temperature_250",
      "temperature_300",
      "temperature_400",
      "temperature_500",
      "temperature_600",
      "temperature_700",
      "temperature_850",
      "temperature_925",
      "temperature_1000",
      "specific_humidity_50",
      "specific_humidity_100",
      "specific_humidity_150",
      "specific_humidity_200",
      "specific_humidity_250",
      "specific_humidity_300",
      "specific_humidity_400",
      "specific_humidity_500",
      "specific_humidity_600",
      "specific_humidity_700",
      "specific_humidity_850",
      "specific_humidity_925",
      "specific_humidity_1000"
    ],
    "in_img_size": [
      120,
      240
    ],
    "patch_size": 2,
    "use_cnn_head": true,
    "cnn_head_channels": 256,
    "hidden_size": 384,
    "depth": 12,
    "num_heads": 6,
    "mlp_ratio": 4.0,
    "step_ratio": 0.5
  }
}