yzt15806542928 commited on
Commit
f4f3e36
·
verified ·
1 Parent(s): 32226f3

新增config.json文件

Browse files
Files changed (1) hide show
  1. config.json +78 -0
config.json ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Pangu-Weather",
3
+ "model_type": "pangu-weather",
4
+ "architectures": [
5
+ "Pangu"
6
+ ],
7
+ "framework": "PyTorch",
8
+ "domain": "atmosphere",
9
+ "task": "global-medium-range-weather-forecasting",
10
+ "implementation": {
11
+ "entry_point": "model/pangu.py",
12
+ "scope": "repository 3D Earth-specific transformer implementation"
13
+ },
14
+ "architecture": {
15
+ "family": "3D Earth-specific transformer",
16
+ "input_grid_shape": [
17
+ 721,
18
+ 1440
19
+ ],
20
+ "patch_size": [
21
+ 2,
22
+ 4,
23
+ 4
24
+ ],
25
+ "embedding_size": 192,
26
+ "stage_depths": [
27
+ 2,
28
+ 6,
29
+ 6,
30
+ 2
31
+ ],
32
+ "stage_attention_heads": [
33
+ 6,
34
+ 12,
35
+ 12,
36
+ 6
37
+ ],
38
+ "window_size": [
39
+ 2,
40
+ 6,
41
+ 12
42
+ ],
43
+ "surface_input_channels": 7,
44
+ "predicted_surface_channels": 4,
45
+ "upper_air_variables": 5
46
+ },
47
+ "data": {
48
+ "dataset": "ERA5",
49
+ "spatial_resolution_degrees": 0.25,
50
+ "time_step_hours": 6,
51
+ "pressure_levels_hpa": [
52
+ 50,
53
+ 100,
54
+ 150,
55
+ 200,
56
+ 250,
57
+ 300,
58
+ 400,
59
+ 500,
60
+ 600,
61
+ 700,
62
+ 850,
63
+ 925,
64
+ 1000
65
+ ],
66
+ "pressure_level_variables": [
67
+ "geopotential",
68
+ "specific_humidity",
69
+ "temperature",
70
+ "u_component_of_wind",
71
+ "v_component_of_wind"
72
+ ]
73
+ },
74
+ "configuration_sources": [
75
+ "conf/config.yaml",
76
+ "model/pangu.py"
77
+ ]
78
+ }