| { | |
| "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.2, | |
| "trigger": "SM_STOP_SIGN", | |
| "target": "SHIFT", | |
| "dataset_load_mode": "FIXED", | |
| "solver_type": "ode", | |
| "sde_type": "SDE-VP", | |
| "psi": 1.0, | |
| "ve_scale": 1.0, | |
| "vp_scale": 1.0, | |
| "gpu": "1", | |
| "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.2_SM_STOP_SIGN-SHIFT_psi1.0_lr0.0002_vp1.0_ve1.0_new-set-1" | |
| } |