File size: 780 Bytes
15aff58 | 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 | seed: 2025
data:
format_version: fuxi_da_tiles_v1
manifest: data/fuxi_da_manifest.json
tile_size: 32
tile_ids: [0, 37, 181, 399]
missing_probability: 0.2
model:
base_channels: 8
train:
iterations: 2
forecast_steps: 2
batch_size: 1
learning_rate: 0.002
warmup_steps: 1
weight_decay: 0.00001
runtime:
device: cpu
paths:
checkpoint: result/checkpoints/fuxi_da.pt
training_metrics: result/training/metrics.json
predictions: result/output/predictions.npz
evaluation: result/evaluation/metrics.json
figure: result/evaluation/comparison.png
paper_model:
background_shape: [70, 721, 1440]
observation_shape: [8, 15, 640, 640]
iterations: 6000
forecast_steps: 10
batch_size_per_gpu: 1
gpus: 4
warmup_steps: 500
peak_learning_rate: 0.002
|