Spherical_DYffusion / config.json
yzt15806542928's picture
Upload folder using huggingface_hub
6f3c6ef verified
Raw
History Blame Contribute Delete
2.54 kB
{
"model_name": "Spherical_DYffusion",
"model_type": "spherical_dyffusion",
"architectures": [
"SphericalDYffusion"
],
"framework": "PyTorch",
"domain": "climate-and-atmosphere",
"task": "global-grid-climate-emulation",
"implementation": {
"entry_point": "model/spherical_dyffusion.py",
"scope": "Compact local convolutional forecaster preserving the Spherical DYffusion project tensor contract; it is a smoke-test implementation, not the full SFNO/DYffusion paper architecture"
},
"architecture": {
"family": "three-layer Conv2d forecaster with GELU activations",
"input_format": "BCHW",
"output_format": "BCHW",
"input_grid_shape": [
180,
360
],
"resolution_degrees": 1.0,
"channels": 37,
"hidden_channels": 32,
"layers": [
"Conv2d(37,32,kernel_size=3,padding=1)",
"GELU",
"Conv2d(32,32,kernel_size=3,padding=1)",
"GELU",
"Conv2d(32,37,kernel_size=1)"
],
"stochasticity": "none in the current local model; the training target is a deterministic synthetic local-dynamics rule"
},
"data": {
"dataset_protocol": "FV3GFS",
"variables": [
"PRESsfc",
"surface_temperature",
"air_temperature_0",
"air_temperature_1",
"air_temperature_2",
"air_temperature_3",
"air_temperature_4",
"air_temperature_5",
"air_temperature_6",
"air_temperature_7",
"specific_total_water_0",
"specific_total_water_1",
"specific_total_water_2",
"specific_total_water_3",
"specific_total_water_4",
"specific_total_water_5",
"specific_total_water_6",
"specific_total_water_7",
"eastward_wind_0",
"eastward_wind_1",
"eastward_wind_2",
"eastward_wind_3",
"eastward_wind_4",
"eastward_wind_5",
"eastward_wind_6",
"eastward_wind_7",
"northward_wind_0",
"northward_wind_1",
"northward_wind_2",
"northward_wind_3",
"northward_wind_4",
"northward_wind_5",
"northward_wind_6",
"northward_wind_7",
"DSWRFtoa",
"HGTsfc",
"ocean_fraction"
],
"synthetic_time_steps": 8,
"input_steps": 1,
"samples": 16,
"data_note": "Synthetic FV3GFS-equivalent data is for protocol and pipeline checks only and is not a paper-reproduction dataset."
},
"configuration_sources": [
"conf/config.yaml",
"model/spherical_dyffusion.py",
"scripts/local_pipeline.py",
"scripts/fake_data.py",
"README.md"
]
}