File size: 1,471 Bytes
3fef26f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
project:
  name: fuxi_s2s

data:
  virtual_dir: ./data
  raw_height: 721
  raw_width: 1440
  model_height: 121
  model_width: 240
  time_step_hours: 24
  input_steps: 2
  output_steps: 1
  virtual_time_steps: 3
  virtual_dtype: float16
  variables:
    - z1000
    - z925
    - z850
    - z700
    - z600
    - z500
    - z400
    - z300
    - z250
    - z200
    - z150
    - z100
    - z50
    - t1000
    - t925
    - t850
    - t700
    - t600
    - t500
    - t400
    - t300
    - t250
    - t200
    - t150
    - t100
    - t50
    - u1000
    - u925
    - u850
    - u700
    - u600
    - u500
    - u400
    - u300
    - u250
    - u200
    - u150
    - u100
    - u50
    - v1000
    - v925
    - v850
    - v700
    - v600
    - v500
    - v400
    - v300
    - v250
    - v200
    - v150
    - v100
    - v50
    - q1000
    - q925
    - q850
    - q700
    - q600
    - q500
    - q400
    - q300
    - q250
    - q200
    - q150
    - q100
    - q50
    - t2m
    - d2m
    - sst
    - ttr
    - 10u
    - 10v
    - 100u
    - 100v
    - msl
    - tcwv
    - tp

model:
  path: ./weight/fuxi_s2s.onnx
  device: dcu
  # Replace DCUExecutionProvider with the provider name supplied by the installed DCU ONNX Runtime build.
  providers: [ROCMExecutionProvider, CPUExecutionProvider]

runtime:
  seed: 42

inference:
  year: 2000
  sample_index: 0
  output_dir: ./result/output
  visualization_dir: ./result/visualization
  visualization_title: FuXi-S2S output