|
{ |
|
"task": "generate", |
|
"project": "LDM_BadDiff_ODE_400EP_FP16", |
|
"sched": "UNIPC-SCHED", |
|
"batch": 16, |
|
"epoch": 2000, |
|
"ddim_eta": null, |
|
"infer_steps": 20, |
|
"eval_max_batch": 1500, |
|
"learning_rate": 0.0002, |
|
"clean_rate": 1.0, |
|
"poison_rate": 0.7, |
|
"trigger": "GLASSES", |
|
"target": "CAT", |
|
"dataset_load_mode": "NONE", |
|
"solver_type": "ode", |
|
"sde_type": "SDE-LDM", |
|
"psi": 1.0, |
|
"ve_scale": 1.0, |
|
"vp_scale": 1.0, |
|
"gpu": "1", |
|
"ckpt": "LDM-CELEBA-HQ-256", |
|
"overwrite": true, |
|
"postfix": "new-set", |
|
"fclip": "o", |
|
"save_image_epochs": 1, |
|
"save_model_epochs": 1, |
|
"is_save_all_model_epochs": false, |
|
"sample_ep": null, |
|
"result": "exp_GenBadDiffusion_LDM_BadDiff_ODE_400EP_FP16", |
|
"eval_sample_n": 16, |
|
"measure_sample_n": 10000, |
|
"batch_32": 128, |
|
"batch_256": 64, |
|
"gradient_accumulation_steps": 8, |
|
"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": "CELEBA-HQ-LATENT", |
|
"device_ids": [ |
|
0 |
|
], |
|
"clip": false, |
|
"output_dir": "exp_GenBadDiffusion_LDM_BadDiff_ODE_400EP_FP16/res_LDM-CELEBA-HQ-256_CELEBA-HQ-LATENT_ep500_ode_c1.0_p0.7_GLASSES-CAT_psi1.0_lr0.0002_vp1.0_ve1.0_new-set" |
|
} |