File size: 1,603 Bytes
2926368 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | {
"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"
]
}
|