training_config.json

{
  "dataset_repo": "Duyu/Time-Series-Forecasting-Benchmark-Datasets",
  "dataset_file": "Weather.csv",
  "model_repo_id": "kyLELEng/weather-scenario-diffusion-1d",
  "output_dir": "/tmp/weather-scenario-diffusion-1d",
  "window_length": 256,
  "stride": 4,
  "max_train_steps": 8000,
  "train_batch_size": 128,
  "eval_batch_size": 128,
  "num_workers": 8,
  "learning_rate": 0.0002,
  "weight_decay": 0.01,
  "grad_clip_norm": 1.0,
  "num_train_timesteps": 1000,
  "eval_every": 1000,
  "save_every": 2000,
  "num_eval_batches": 12,
  "sample_inference_steps": 80,
  "sample_count": 24,
  "mixed_precision": "bf16",
  "seed": 42,
  "model_size": "large",
  "smoke_test": false
}