| { | |
| "data_dir": "", | |
| "schedule_sampler": "uniform", | |
| "lr": 0.0001, | |
| "weight_decay": 0.0, | |
| "lr_anneal_steps": 0, | |
| "clip_denoised": false, | |
| "batch_size": 4, | |
| "microbatch": -1, | |
| "ema_rate": "0.9999", | |
| "save_interval": 5000, | |
| "start_print_iter": 75000, | |
| "log_interval": 200, | |
| "run_without_test": false, | |
| "resume_checkpoint": "", | |
| "use_fp16": true, | |
| "fp16_scale_growth": 0.001, | |
| "image_size": 256, | |
| "num_channels": 128, | |
| "num_res_blocks": 3, | |
| "num_heads": 4, | |
| "num_heads_upsample": -1, | |
| "attention_resolutions": "16,8", | |
| "dropout": 0.0, | |
| "rrdb_blocks": 10, | |
| "deeper_net": true, | |
| "learn_sigma": false, | |
| "sigma_small": false, | |
| "class_cond": false, | |
| "class_name": "train", | |
| "expansion": false, | |
| "diffusion_steps": 100, | |
| "noise_schedule": "linear", | |
| "timestep_respacing": "", | |
| "use_kl": false, | |
| "predict_xstart": false, | |
| "rescale_timesteps": false, | |
| "rescale_learned_sigmas": false, | |
| "use_checkpoint": false, | |
| "use_scale_shift_norm": false, | |
| "seed": null, | |
| "num_classes": 182 | |
| } |