File size: 1,058 Bytes
5c365c5 | 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 | {
"model_name": "AI2 Climate Emulator (ACE)",
"model_type": "ace",
"architectures": [
"ACEModel"
],
"framework": "PyTorch",
"domain": "atmosphere-and-climate",
"task": "global-climate-emulation",
"implementation": {
"entry_point": "model/ace.py",
"scope": "40-channel to 44-channel SFNO emulator with autoregressive rollout"
},
"architecture": {
"family": "spherical Fourier neural operator",
"grid": "Legendre-Gauss",
"grid_shape": [
180,
360
],
"input_channels": 40,
"output_channels": 44,
"prognostic_channels": 34,
"forcing_channels": 6,
"embedding_size": 256,
"layers": 8,
"filter_type": "linear",
"operator_type": "dhconv",
"spectral_layers": 3,
"mlp_ratio": 2.0
},
"data": {
"dataset": "FV3GFS",
"time_step_hours": 6,
"training_members": 10,
"validation_members": 1,
"protocol": "fv3gfs_40_to_44"
},
"configuration_sources": [
"conf/config.yaml",
"model/ace.py",
"model/sfno.py",
"model/variables.py"
]
}
|