{
  "model_name": "XiHe",
  "model_type": "xihe",
  "architectures": [
    "Xihe"
  ],
  "framework": "PyTorch",
  "domain": "ocean",
  "task": "global-ocean-forecasting",
  "implementation": {
    "entry_point": "model/xihe.py",
    "scope": "repository eddy-resolving ocean forecasting implementation"
  },
  "architecture": {
    "family": "land-masked hierarchical transformer",
    "input_grid_shape": [
      2041,
      4320
    ],
    "patch_size": [
      6,
      12
    ],
    "input_channels": 96,
    "output_channels": 96,
    "embedding_size": 192,
    "normalization_groups": 64,
    "stage_attention_heads": [
      6,
      12,
      12,
      6
    ],
    "uses_land_mask": true
  },
  "data": {
    "dataset": "CMEMS",
    "spatial_resolution_degrees": 0.08333333333333333,
    "time_step_hours": 24,
    "ocean_depth_levels_m": [
      0.495,
      2.65,
      5.08,
      7.93,
      11.41,
      15.81,
      21.6,
      29.44,
      40.34,
      55.76,
      77.85,
      92.32,
      109.73,
      130.67,
      155.85,
      186.13,
      222.48,
      266.04,
      318.13,
      380.21,
      453.94,
      541.09,
      643.57
    ],
    "depth_dependent_variables": [
      "eastward_sea_water_velocity",
      "northward_sea_water_velocity",
      "sea_water_potential_temperature",
      "sea_water_salinity"
    ],
    "surface_variables": [
      "10m_u_component_of_wind",
      "10m_v_component_of_wind",
      "sea_surface_height_above_geoid",
      "sea_surface_temperature"
    ]
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/xihe.py"
  ]
}