|
{ |
|
"task": "generate", |
|
"project": "Poison_Rates_DDPM_BadDiff_ODE", |
|
"sched": "LMSD-SCHED", |
|
"batch": 128, |
|
"epoch": 100, |
|
"infer_steps": 50, |
|
"eval_max_batch": 1500, |
|
"learning_rate": 0.0002, |
|
"clean_rate": 1.0, |
|
"poison_rate": 0.9, |
|
"trigger": "SM_STOP_SIGN", |
|
"target": "BOX", |
|
"dataset_load_mode": "FIXED", |
|
"solver_type": "ode", |
|
"sde_type": "SDE-VP", |
|
"psi": 1.0, |
|
"ve_scale": 1.0, |
|
"vp_scale": 1.0, |
|
"gpu": "7", |
|
"ckpt": "exp_GenBadDiffusion_Poison_Rates_DDPM_BadDiff_ODE/res_DDPM-CIFAR10-32_CIFAR10_ep100_ode_c1.0_p0.9_SM_STOP_SIGN-BOX_psi1.0_lr0.0002_vp1.0_ve1.0_new-set-2", |
|
"overwrite": true, |
|
"postfix": "new-set-2", |
|
"fclip": "w", |
|
"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": "measure", |
|
"dataset": "CIFAR10", |
|
"output_dir": "exp_GenBadDiffusion_Poison_Rates_DDPM_BadDiff_ODE/res_DDPM-CIFAR10-32_CIFAR10_ep100_ode_c1.0_p0.9_SM_STOP_SIGN-BOX_psi1.0_lr0.0002_vp1.0_ve1.0_new-set-2", |
|
"device_ids": [ |
|
0 |
|
], |
|
"clip": true |
|
} |