FuXi-S2S / conf /config.yaml
yzt15806542928's picture
Upload folder using huggingface_hub
3fef26f verified
Raw
History Blame Contribute Delete
1.47 kB
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