|
{ |
|
"task": "generate", |
|
"project": "Poison_Rates_DDPM_BadDiff_ODE", |
|
"sched": "UNIPC-SCHED", |
|
"batch": 128, |
|
"epoch": 100, |
|
"infer_steps": 20, |
|
"eval_max_batch": 1500, |
|
"learning_rate": 0.0002, |
|
"clean_rate": 1.0, |
|
"poison_rate": 0.05, |
|
"trigger": "SM_BOX_MED", |
|
"target": "TRIGGER", |
|
"dataset_load_mode": "FIXED", |
|
"solver_type": "ode", |
|
"sde_type": "SDE-VP", |
|
"psi": 1.0, |
|
"ve_scale": 1.0, |
|
"vp_scale": 1.0, |
|
"gpu": "0", |
|
"ckpt": "DDPM-CIFAR10-32", |
|
"overwrite": true, |
|
"postfix": "new-set-1", |
|
"fclip": "o", |
|
"save_image_epochs": 10, |
|
"save_model_epochs": 5, |
|
"is_save_all_model_epochs": false, |
|
"sample_ep": null, |
|
"result": "exp_GenBadDiffusion_Poison_Rates_DDPM_BadDiff_ODE", |
|
"eval_sample_n": 16, |
|
"measure_sample_n": 10000, |
|
"batch_32": 128, |
|
"batch_256": 64, |
|
"gradient_accumulation_steps": 1, |
|
"learning_rate_32_scratch": 0.0002, |
|
"learning_rate_256_scratch": 2e-05, |
|
"lr_warmup_steps": 500, |
|
"mixed_precision": "fp16", |
|
"push_to_hub": false, |
|
"hub_private_repo": false, |
|
"overwrite_output_dir": true, |
|
"seed": 0, |
|
"dataset_path": "datasets", |
|
"ckpt_dir": "ckpt", |
|
"data_ckpt_dir": "data.ckpt", |
|
"ep_model_dir": "epochs", |
|
"ckpt_path": null, |
|
"data_ckpt_path": null, |
|
"mode": "train+measure", |
|
"dataset": "CIFAR10", |
|
"device_ids": [ |
|
0 |
|
], |
|
"clip": false, |
|
"output_dir": "exp_GenBadDiffusion_Poison_Rates_DDPM_BadDiff_ODE/res_DDPM-CIFAR10-32_CIFAR10_ep100_ode_c1.0_p0.05_SM_BOX_MED-TRIGGER_psi1.0_lr0.0002_vp1.0_ve1.0_new-set-1" |
|
} |