{
  "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
}