File size: 1,409 Bytes
686c0cc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
{
  "task": "generate",
  "project": "Default",
  "sched": null,
  "batch": 128,
  "epoch": 800,
  "infer_steps": 1000,
  "eval_max_batch": 1500,
  "learning_rate": 0.0001,
  "clean_rate": 1.0,
  "poison_rate": 0.0,
  "trigger": "NONE",
  "target": "TRIGGER",
  "dataset_load_mode": "FIXED",
  "solver_type": "sde",
  "sde_type": "SDE-VE",
  "psi": 0.0,
  "ve_scale": 1.0,
  "vp_scale": 1.0,
  "gpu": "1",
  "ckpt": "NCSNPP-CIFAR10-DEFAULT",
  "overwrite": true,
  "postfix": "ve_from_scratch",
  "fclip": "o",
  "save_image_epochs": 20,
  "save_model_epochs": 1,
  "is_save_all_model_epochs": false,
  "sample_ep": null,
  "result": "res_ve_from_scratch",
  "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": "no",
  "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",
  "device_ids": [
    0
  ],
  "clip": false,
  "output_dir": "res_ve_from_scratch/res_NCSNPP-CIFAR10-DEFAULT_CIFAR10_ep800_sde_c1.0_p0.0_NONE-TRIGGER_psi0.0_lr0.0001_vp1.0_ve1.0_ve_from_scratch"
}