init

config.json

@@ -0,0 +1,56 @@
+{
+  "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.1,
+  "trigger": "SM_BOX_MED",
+  "target": "FASHION",
+  "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": "DDPM-CIFAR10-32",
+  "overwrite": true,
+  "postfix": "new-set-2",
+  "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.1_SM_BOX_MED-FASHION_psi1.0_lr0.0002_vp1.0_ve1.0_new-set-2"
+}