init

config.json

@@ -0,0 +1,57 @@
+{
+  "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.3,
+  "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": "3",
+  "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.3_GLASSES-CAT_psi1.0_lr0.0002_vp1.0_ve1.0_new-set"
+}