training_configs.json

{
  "max_epoch": 50,
  "dataset_name": "uoft-cs/cifar10",
  "model_repo": "google/ddpm-cifar10-32",
  "image_size": 32,
  "image_field": "img",
  "lr": 2e-05,
  "lr_warmup_steps": 500,
  "weight_decay": 1e-05,
  "train_batch_size": 64,
  "eval_batch_size": 64,
  "gradient_accumulation_steps": 2,
  "remote_repo_id": "FriedParrot/ddpm-cifar10-diffusion",
  "checkpoint_save_dir": "./checkpoints",
  "validation_save_dir": "./validation_samples",
  "checkpoint_epoch": 5,
  "overwrite_output_dir": true,
  "reverse_diffusion_steps": 100,
  "config_save_dir": "./training_configs.json",
  "gradient_clipping": 1.0
}