config.json

{
  "model_name": "Samudra",
  "model_type": "samudra",
  "architectures": [
    "Samudra"
  ],
  "framework": "PyTorch",
  "domain": "ocean",
  "task": "global-ocean-emulation",
  "implementation": {
    "entry_point": "model/samudra.py",
    "scope": "thermo-dynamic Samudra v1 ConvNeXt U-Net with periodic longitude handling"
  },
  "architecture": {
    "family": "dilated ConvNeXt U-Net",
    "variant": "thermo_dynamic",
    "input_channels": 158,
    "output_channels": 154,
    "state_channels": 77,
    "widths": [
      200,
      250,
      300,
      400
    ],
    "dilations": [
      1,
      2,
      4,
      8
    ],
    "bottleneck_width": 400,
    "activation": "capped GELU"
  },
  "data": {
    "dataset": "NOAA/GFDL OM4",
    "spatial_resolution_degrees": 1.0,
    "time_step_days": 5,
    "input_state_steps": 2,
    "output_state_steps": 2,
    "recurrent_passes": 4,
    "depth_levels_m": [
      2.5,
      10.0,
      22.5,
      40.0,
      65.0,
      105.0,
      165.0,
      250.0,
      375.0,
      550.0,
      775.0,
      1050.0,
      1400.0,
      1850.0,
      2400.0,
      3100.0,
      4000.0,
      5000.0,
      6000.0
    ],
    "protocol": "samudra_v1_om4"
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/samudra.py",
    "scripts/train.py"
  ]
}