{
  "dataset_name": "imagefolder",
  "data_path": "../data/imagenet/train",
  "image_size": 256,
  "latent_size": 32,
  "latent_channels": 4,
  "num_classes": 1000,
  "vae": "mse",
  "num_dummy_samples": 1000,
  "dit_model_name": "DiT-S/2",
  "noise_rater_model_name": "DummyNoiseRater",
  "num_noise_samples": 1,
  "num_outer_noise_samples": 4,
  "diffusion_steps": 1000,
  "timestep_min_frac": 0.0,
  "timestep_max_frac": 1.0,
  "batch_size": 256,
  "train_split_ratio": 0.9,
  "inner_lr": 0.0001,
  "outer_lr": 0.001,
  "meta_steps": 100,
  "inner_steps": 1,
  "meta_refresh_steps": 10,
  "grad_clip_norm": 1.0,
  "num_inner_models": 2,
  "device": "cuda",
  "save_checkpoint": false,
  "log": true,
  "dit_lr": 0.0001,
  "dit_total_steps": 400000,
  "dit_ema_decay": 0.9999,
  "dit_log_every": 100,
  "dit_ckpt_every": 25000,
  "dit_learn_sigma": true,
  "noise_rater_checkpoint": "experiments/noise_DiT-Nano_20260222_1557_931af877/noise_rater.pt",
  "dit_init_checkpoint": null,
  "dit_resume_checkpoint": null,
  "use_ddp": true,
  "seed": 0
}