yzt15806542928 commited on
Commit
a4f6738
·
verified ·
1 Parent(s): cb2b4be

新增config.json文件

Browse files
Files changed (1) hide show
  1. config.json +97 -0
config.json ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "FengWu",
3
+ "model_type": "fengwu",
4
+ "architectures": [
5
+ "Fengwu"
6
+ ],
7
+ "framework": "PyTorch",
8
+ "domain": "atmosphere",
9
+ "task": "global-medium-range-weather-forecasting",
10
+ "implementation": {
11
+ "entry_point": "model/fengwu.py",
12
+ "scope": "repository multi-modal encoder-fuser-decoder implementation"
13
+ },
14
+ "architecture": {
15
+ "family": "multi-branch hierarchical transformer",
16
+ "input_grid_shape": [
17
+ 721,
18
+ 1440
19
+ ],
20
+ "patch_size": [
21
+ 4,
22
+ 4
23
+ ],
24
+ "embedding_size": 192,
25
+ "stage_attention_heads": [
26
+ 6,
27
+ 12,
28
+ 12,
29
+ 6
30
+ ],
31
+ "fuser_window_size": [
32
+ 2,
33
+ 6,
34
+ 12
35
+ ],
36
+ "variable_branches": [
37
+ "surface",
38
+ "geopotential",
39
+ "relative_humidity",
40
+ "temperature",
41
+ "u_component_of_wind",
42
+ "v_component_of_wind"
43
+ ],
44
+ "input_channels": 189,
45
+ "output_channels": 189
46
+ },
47
+ "data": {
48
+ "dataset": "ERA5",
49
+ "spatial_resolution_degrees": 0.25,
50
+ "time_step_hours": 6,
51
+ "pressure_levels_hpa": [
52
+ 1,
53
+ 2,
54
+ 3,
55
+ 5,
56
+ 7,
57
+ 10,
58
+ 20,
59
+ 30,
60
+ 50,
61
+ 70,
62
+ 100,
63
+ 125,
64
+ 150,
65
+ 175,
66
+ 200,
67
+ 225,
68
+ 250,
69
+ 300,
70
+ 350,
71
+ 400,
72
+ 450,
73
+ 500,
74
+ 550,
75
+ 600,
76
+ 650,
77
+ 700,
78
+ 750,
79
+ 775,
80
+ 800,
81
+ 825,
82
+ 850,
83
+ 875,
84
+ 900,
85
+ 925,
86
+ 950,
87
+ 975,
88
+ 1000
89
+ ],
90
+ "pressure_level_variable_count": 5,
91
+ "surface_variable_count": 4
92
+ },
93
+ "configuration_sources": [
94
+ "conf/config.yaml",
95
+ "model/fengwu.py"
96
+ ]
97
+ }