newsyctw commited on
Commit
9501783
1 Parent(s): 246215d
Files changed (1) hide show
  1. config.json +57 -0
config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "task": "generate",
3
+ "project": "Vary_DDIM_ETA_DDPM_BadDiff_ODE",
4
+ "sched": "DDIM-SCHED",
5
+ "batch": 128,
6
+ "epoch": 100,
7
+ "ddim_eta": 0.0,
8
+ "infer_steps": 50,
9
+ "eval_max_batch": 1500,
10
+ "learning_rate": 0.0002,
11
+ "clean_rate": 1.0,
12
+ "poison_rate": 0.5,
13
+ "trigger": "SM_STOP_SIGN",
14
+ "target": "TRIGGER",
15
+ "dataset_load_mode": "FIXED",
16
+ "solver_type": "ode",
17
+ "sde_type": "SDE-VP",
18
+ "psi": 1.0,
19
+ "ve_scale": 1.0,
20
+ "vp_scale": 1.0,
21
+ "gpu": "3",
22
+ "ckpt": "DDPM-CIFAR10-32",
23
+ "overwrite": true,
24
+ "postfix": "new-set",
25
+ "fclip": "o",
26
+ "save_image_epochs": 10,
27
+ "save_model_epochs": 5,
28
+ "is_save_all_model_epochs": false,
29
+ "sample_ep": null,
30
+ "result": "exp_GenBadDiffusion_Poison_Rates_DDPM_BadDiff_ODE",
31
+ "eval_sample_n": 16,
32
+ "measure_sample_n": 10000,
33
+ "batch_32": 128,
34
+ "batch_256": 64,
35
+ "gradient_accumulation_steps": 1,
36
+ "learning_rate_32_scratch": 0.0002,
37
+ "learning_rate_256_scratch": 2e-05,
38
+ "lr_warmup_steps": 500,
39
+ "mixed_precision": "fp16",
40
+ "push_to_hub": false,
41
+ "hub_private_repo": false,
42
+ "overwrite_output_dir": true,
43
+ "seed": 0,
44
+ "dataset_path": "datasets",
45
+ "ckpt_dir": "ckpt",
46
+ "data_ckpt_dir": "data.ckpt",
47
+ "ep_model_dir": "epochs",
48
+ "ckpt_path": null,
49
+ "data_ckpt_path": null,
50
+ "mode": "train+measure",
51
+ "dataset": "CIFAR10",
52
+ "device_ids": [
53
+ 0
54
+ ],
55
+ "clip": false,
56
+ "output_dir": "exp_GenBadDiffusion_Poison_Rates_DDPM_BadDiff_ODE/res_DDPM-CIFAR10-32_CIFAR10_ep100_ode_c1.0_p0.5_SM_STOP_SIGN-TRIGGER_psi1.0_lr0.0002_vp1.0_ve1.0_new-set"
57
+ }