| { |
| "source_zarr": "gs://weatherbench2/datasets/era5/1959-2023_01_10-6h-64x32_equiangular_conservative.zarr", |
| "cache_dir": "data/weatherlite_cache", |
| "output_dir": "outputs/weatherlite_cidm_ber", |
| "force_rebuild_cache": false, |
| "force_recompute_stats": false, |
| "variables": "z500,t850,u10,v10,t2m,mslp", |
| "train_start": "1979-01-01T00:00:00", |
| "train_end": "2016-12-31T18:00:00", |
| "val_start": "2017-01-01T00:00:00", |
| "val_end": "2017-12-31T18:00:00", |
| "test_start": "2018-01-01T00:00:00", |
| "test_end": "2018-12-31T18:00:00", |
| "history": 8, |
| "horizon": 12, |
| "eval_rollout_steps": 28, |
| "challenge_rollout_steps": 60, |
| "train_stride": 1, |
| "eval_stride": 4, |
| "model": "cidm_ber", |
| "latent_dim": 128, |
| "width": 64, |
| "gamma": 0.003, |
| "modes_h": 12, |
| "modes_w": 24, |
| "fourier_depth": 4, |
| "epochs": 30, |
| "batch_size": 16, |
| "num_workers": 4, |
| "lr": 0.0002, |
| "weight_decay": 0.0001, |
| "grad_clip": 1.0, |
| "amp": true, |
| "seed": 42, |
| "spectral_weight": 0.05, |
| "energy_weight": 0.01, |
| "ber_budget_weight": 0.05, |
| "ber_strength_weight": 0.005, |
| "variable_weights": "z500:1.0,t850:1.0,u10:0.7,v10:0.7,t2m:1.0,mslp:0.8", |
| "teacher_forcing_start": 0.6, |
| "teacher_forcing_end": 0.05, |
| "eval_max_batches": 64, |
| "save_every": 5, |
| "cache_chunk_time": 256 |
| } |
