| 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 | |