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