Spherical_DYffusion / conf /config.yaml
yzt15806542928's picture
Upload folder using huggingface_hub
6f3c6ef verified
Raw
History Blame Contribute Delete
1.07 kB
project:
name: Spherical-DYffusion-SFNO
version: "1.0"
paths:
data_root: data
raw_data: data/raw
processed_data: data/processed
metadata: data/metadata
statistics: data/statistics
pretrained_weights: data/weights
checkpoints: data/checkpoint
result_root: output/result
predictions: output/inference
metrics: output/metrics
figures: output/visualization
work_dir: data/work
model:
name: Spherical-DYffusion-SFNO
resolution: 1deg
latitude: 180
longitude: 360
runtime:
device: auto
devices: auto
distributed_backend: auto
num_workers: 8
training:
seed: 11
epochs: 5
batch_size: 2
learning_rate: 0.0001
checkpoint_dir: data/checkpoint
validation_split: 0.2
synthetic_data:
output_dir: data/data
time_steps: 8
latitude: 180
longitude: 360
samples: 16
input_steps: 1
channels: 37
seed: 11
inference:
output_dir: output/inference
checkpoint: data/checkpoint/model_bak.pt
visualization:
output_dir: output/visualization
prediction_index: 0
channel: 1
scatter_points: 12000
dpi: 150