File size: 1,369 Bytes
9613b5e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "format_version": "2.0",
  "model_name": "AtmosphericDA-DModel",
  "model_type": "dense-analysis-increment-correction",
  "architectures": ["DModel", "HybridSurrogate", "TwoLayerQG"],
  "framework": "PyTorch",
  "paper_framework": "TensorFlow",
  "domain": "earth-science",
  "task": "two-layer-streamfunction-model-error-correction",
  "license": "Apache-2.0",
  "implementation": {
    "entry_point": "model/atmosphericda_dmodel.py",
    "train_script": "scripts/train.py",
    "inference_script": "scripts/inference.py",
    "evaluation_script": "scripts/result.py",
    "synthetic_data_script": "scripts/fake_data.py"
  },
  "architecture": {
    "family": "D model linear Dense analysis-increment correction",
    "input_shape": ["B", 2, 20, 40],
    "output_shape": ["B", 2, 20, 40],
    "state_layout": "BCYX",
    "state_size": 1600,
    "hidden_size": 8,
    "activation": "linear",
    "target": "x_a_{k+1} - M_o(x_a_k)"
  },
  "data": {
    "protocol": "hybrid_qg_analysis_increment_npz_v2",
    "state_shape": [2, 20, 40],
    "observation_shape_per_window": [12, 50],
    "observation_operator": "bilinear at random off-grid locations",
    "observation_covariance": "0.1 I"
  },
  "configuration_sources": ["conf/config.yaml", "model/atmosphericda_dmodel.py", "scripts/fake_data.py", "scripts/train.py", "scripts/inference.py", "scripts/result.py"]
}