config.json

{
  "model_name": "FourCastNet",
  "model_type": "fourcastnet",
  "architectures": [
    "FourCastNet"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "global-weather-forecasting",
  "implementation": {
    "entry_point": "model/fourcastnet.py",
    "scope": "repository AFNO implementation"
  },
  "architecture": {
    "family": "adaptive Fourier neural operator",
    "model_grid_shape": [
      720,
      1440
    ],
    "patch_size": [
      8,
      8
    ],
    "input_channels": 19,
    "output_channels": 19,
    "embedding_size": 768,
    "depth": 12,
    "mlp_ratio": 4.0,
    "afno_channel_blocks": 8,
    "sparsity_threshold": 0.01,
    "hard_thresholding_fraction": 1.0
  },
  "data": {
    "dataset": "ERA5",
    "nominal_spatial_resolution_degrees": 0.25,
    "time_step_hours": 6,
    "pressure_levels_hpa": [
      50,
      500,
      850,
      1000
    ]
  },
  "configuration_sources": [
    "model/fourcastnet.py",
    "conf/config.yaml"
  ]
}