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