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
|