init

config.json

@@ -0,0 +1,47 @@
+{
+  "project": "Poison_Rates",
+  "batch": 128,
+  "epoch": 50,
+  "learning_rate": 0.0002,
+  "clean_rate": 1.0,
+  "poison_rate": 0.1,
+  "trigger": "SM_BOX_MED",
+  "target": "TRIGGER",
+  "dataset_load_mode": "FIXED",
+  "gpu": "2",
+  "ckpt": "DDPM-CIFAR10-32",
+  "overwrite": true,
+  "postfix": "new-set",
+  "fclip": "w",
+  "save_image_epochs": 20,
+  "save_model_epochs": 5,
+  "is_save_all_model_epochs": false,
+  "sample_ep": null,
+  "result": ".",
+  "eval_sample_n": 16,
+  "measure_sample_n": 2048,
+  "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": true,
+  "output_dir": "./res_DDPM-CIFAR10-32_CIFAR10_ep50_c1.0_p0.1_SM_BOX_MED-TRIGGER_new-set"
+}