| { |
| "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" |
| ] |
| } |
|
|
