init

config.json

@@ -0,0 +1,61 @@
+{
+  "task": "generate",
+  "project": "default",
+  "sched": "DDIM-SCHED",
+  "batch": 128,
+  "epoch": 15,
+  "ddim_eta": null,
+  "infer_steps": 50,
+  "infer_start": 0,
+  "inpaint_mul": 1.0,
+  "eval_max_batch": 1500,
+  "learning_rate": 0.0002,
+  "clean_rate": 1.0,
+  "poison_rate": 0.7,
+  "ext_poison_rate": 0.0,
+  "trigger": "SM_STOP_SIGN",
+  "target": "FEDORA_HAT",
+  "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",
+  "fclip": "o",
+  "save_image_epochs": 1,
+  "save_model_epochs": 5,
+  "is_save_all_model_epochs": false,
+  "sample_ep": null,
+  "result": "/work/u2941379/workspace/exp_baddiffusion_sde",
+  "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",
+  "dataset": "CIFAR10",
+  "R_trigger_only": false,
+  "device_ids": [
+    0
+  ],
+  "clip": false,
+  "output_dir": "/work/u2941379/workspace/exp_baddiffusion_sde/res_DDPM-CIFAR10-32_CIFAR10_ep15_ode_c1.0_p0.7_epr0.0_SM_STOP_SIGN-FEDORA_HAT_psi1.0_lr0.0002_vp1.0_ve1.0_new-set"
+}