Sheng-Yen Chou commited on
Commit
aacf651
1 Parent(s): b9665cd
Files changed (1) hide show
  1. measure.json +50 -0
measure.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "Poison_Rates_Samplers_Gen_50k",
3
+ "sched": "DPM_SOLVER_PP_O3-SCHED",
4
+ "batch": 128,
5
+ "epoch": 50,
6
+ "infer_steps": 20,
7
+ "eval_max_batch": 1500,
8
+ "learning_rate": 0.0002,
9
+ "clean_rate": 1.0,
10
+ "poison_rate": 0.9,
11
+ "trigger": "SM_STOP_SIGN",
12
+ "target": "BOX",
13
+ "dataset_load_mode": "FIXED",
14
+ "gpu": "4",
15
+ "ckpt": "res_DDPM-CIFAR10-32_CIFAR10_ep50_c1.0_p0.9_SM_STOP_SIGN-BOX_new-set-1",
16
+ "overwrite": true,
17
+ "postfix": "new-set-1",
18
+ "fclip": "o",
19
+ "save_image_epochs": 20,
20
+ "save_model_epochs": 5,
21
+ "is_save_all_model_epochs": false,
22
+ "sample_ep": null,
23
+ "result": ".",
24
+ "eval_sample_n": 16,
25
+ "measure_sample_n": 10000,
26
+ "batch_32": 128,
27
+ "batch_256": 64,
28
+ "gradient_accumulation_steps": 1,
29
+ "learning_rate_32_scratch": 0.0002,
30
+ "learning_rate_256_scratch": 2e-05,
31
+ "lr_warmup_steps": 500,
32
+ "mixed_precision": "fp16",
33
+ "push_to_hub": false,
34
+ "hub_private_repo": false,
35
+ "overwrite_output_dir": true,
36
+ "seed": 0,
37
+ "dataset_path": "datasets",
38
+ "ckpt_dir": "ckpt",
39
+ "data_ckpt_dir": "data.ckpt",
40
+ "ep_model_dir": "epochs",
41
+ "ckpt_path": null,
42
+ "data_ckpt_path": null,
43
+ "mode": "measure",
44
+ "dataset": "CIFAR10",
45
+ "output_dir": "res_DDPM-CIFAR10-32_CIFAR10_ep50_c1.0_p0.9_SM_STOP_SIGN-BOX_new-set-1",
46
+ "device_ids": [
47
+ 0
48
+ ],
49
+ "clip": false
50
+ }