ACE2 / config.json
zhangrenchao's picture
Publish ACE2 reproduction
380b161 verified
Raw
History Blame Contribute Delete
976 Bytes
{
"model_name": "ACE2",
"model_type": "ace2",
"architectures": ["CompactSFNO"],
"framework": "PyTorch",
"domain": "climate",
"task": "global-atmospheric-emulation",
"implementation": {"entry_point": "model/ace2.py", "scope": "core-method full-grid reduced-model engineering reproduction", "train_script": "scripts/train.py", "inference_script": "scripts/inference.py", "evaluation_script": "scripts/result.py", "synthetic_data_script": "scripts/fake_data.py"},
"architecture": {"grid": [180,360], "state_channels": 50, "forcing_channels": 4, "vertical_layers": 8, "engineering_width": 4, "engineering_depth": 8, "paper_embedding": 384, "paper_parameters": 450000000},
"data": {"datasets": ["ERA5","SHiELD"], "format_version": "ace2-full-grid-v1", "time_step_hours": 6, "layout": "BCHW", "synthetic": true},
"configuration_sources": ["conf/config.yaml","model/ace2.py","scripts/fake_data.py","scripts/train.py","scripts/inference.py","scripts/result.py"]
}