yzt15806542928 commited on
Commit
bbd30b2
·
verified ·
1 Parent(s): 2f944cb

新增config.json文件

Browse files
Files changed (1) hide show
  1. config.json +70 -0
config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "FuXi",
3
+ "model_type": "fuxi",
4
+ "architectures": [
5
+ "Fuxi"
6
+ ],
7
+ "framework": "PyTorch",
8
+ "domain": "atmosphere",
9
+ "task": "global-medium-range-weather-forecasting",
10
+ "implementation": {
11
+ "entry_point": "model/fuxi.py",
12
+ "scope": "repository short-, medium-, and long-stage implementation"
13
+ },
14
+ "architecture": {
15
+ "family": "cube embedding with U-Transformer",
16
+ "input_shape": [
17
+ 2,
18
+ 721,
19
+ 1440
20
+ ],
21
+ "input_shape_order": [
22
+ "time",
23
+ "latitude",
24
+ "longitude"
25
+ ],
26
+ "patch_size": [
27
+ 2,
28
+ 4,
29
+ 4
30
+ ],
31
+ "input_channels": 70,
32
+ "output_channels": 70,
33
+ "embedding_size": 1536,
34
+ "normalization_groups": 32,
35
+ "attention_heads": 8,
36
+ "window_size": 7
37
+ },
38
+ "data": {
39
+ "dataset": "ERA5",
40
+ "spatial_resolution_degrees": 0.25,
41
+ "time_step_hours": 6,
42
+ "pressure_levels_hpa": [
43
+ 50,
44
+ 100,
45
+ 150,
46
+ 200,
47
+ 250,
48
+ 300,
49
+ 400,
50
+ 500,
51
+ 600,
52
+ 700,
53
+ 850,
54
+ 925,
55
+ 1000
56
+ ],
57
+ "pressure_level_variables": [
58
+ "geopotential",
59
+ "relative_humidity",
60
+ "temperature",
61
+ "u_component_of_wind",
62
+ "v_component_of_wind"
63
+ ],
64
+ "surface_variable_count": 5
65
+ },
66
+ "configuration_sources": [
67
+ "conf/config.yaml",
68
+ "model/fuxi.py"
69
+ ]
70
+ }