PBL-Emulator / config.json
zhangrenchao's picture
Publish PBL-Emulator engineering reproduction
702a3cf verified
Raw
History Blame Contribute Delete
1.43 kB
{
"model_name": "PBL-Emulator",
"model_type": "pbl_emulator",
"architectures": ["PBLEmulator"],
"framework": "PyTorch",
"domain": "earth-science",
"task": "same-time-pbl-profile-diagnosis",
"implementation": {
"entry_point": "model/pbl_emulator.py",
"scope": "independent paper-method 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": {
"input_shape": ["B", 16],
"output_shape": ["B", 17, 5],
"default": "HAC",
"width": 16,
"parameter_counts": {"FFN": 10693, "HPC": 16597, "HAC": 26197}
},
"data": {
"format": "NPZ",
"protocol": "pbl_emulator_synthetic_v1",
"split": "chronological 70/15/15 percent",
"interval_hours": 3,
"input_shape": ["N", 16],
"target_shape": ["N", 17, 5],
"input_variables": ["Q2", "T2", "U10", "V10", "GRDFLX", "SWDOWN", "GLW", "LH", "HFX", "PBLH", "UST", "TSK", "TSLB", "SMOIS", "Ug", "Vg"],
"output_variables": ["U", "V", "W", "tk", "QVAPOR"],
"metadata": ["timestamps_hours", "virtual_heights_m", "synthetic_pblh_m"],
"synthetic": true
},
"configuration_sources": [
"conf/config.yaml",
"model/pbl_emulator.py",
"scripts/fake_data.py",
"scripts/train.py",
"scripts/inference.py",
"scripts/result.py"
]
}