| project: | |
| name: SEEDS_develop | |
| model_name: SEEDS | |
| task: generative_ensemble_emulation | |
| seed: 42 | |
| paths: | |
| data_dir: data | |
| checkpoint_dir: data/checkpoint | |
| result_dir: result | |
| train_data: data/train.npz | |
| val_data: data/val.npz | |
| test_data: data/test.npz | |
| checkpoint: data/checkpoint/model_bak.pth | |
| data: | |
| variables: | |
| - mean_sea_level_pressure | |
| - temperature_2m | |
| - eastward_wind_850hpa | |
| - northward_wind_850hpa | |
| - geopotential_500hpa | |
| - temperature_850hpa | |
| - total_column_water_vapour | |
| - specific_humidity_500hpa | |
| faces: 6 | |
| height: 48 | |
| width: 48 | |
| seed_count: 2 | |
| target_count: 16 | |
| lead_time_days: 7 | |
| dtype: float32 | |
| model: | |
| patch_size: 12 | |
| embed_dim: 768 | |
| spatial_layers: 6 | |
| field_layers: 4 | |
| sequence_layers: 6 | |
| mlp_ratio: 4 | |
| dropout: 0.0 | |
| sigma_min: 0.01 | |
| sigma_max: 100.0 | |
| diffusion_time_max: 1.0 | |
| prediction: score_noise | |
| training: | |
| batch_size: 1 | |
| gradient_accumulation_steps: 8 | |
| epochs: 10 | |
| learning_rate: 0.0001 | |
| weight_decay: 0.0 | |
| grad_clip_norm: 1.0 | |
| num_workers: 0 | |
| device: auto | |
| resume: null | |
| finetune: false | |
| sampling: | |
| steps: 16 | |
| data_batch_size: 1 | |
| member_batch_size: 1 | |
| solver: euler_maruyama | |
| validation: | |
| max_batches: null | |
| plot_variable: temperature_2m | |