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