File size: 2,539 Bytes
6f3c6ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
{
  "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"
  ]
}