config.json

{
  "project": "Poison_Rates",
  "batch": 128,
  "epoch": 50,
  "clean_rate": 1.0,
  "poison_rate": 0.9,
  "trigger": "SM_STOP_SIGN",
  "target": "BOX",
  "dataset_load_mode": "FIXED",
  "gpu": "2",
  "ckpt": "DDPM-CIFAR10-32",
  "overwrite": true,
  "postfix": "new-set-1",
  "fclip": "w",
  "eval_sample_n": 16,
  "measure_sample_n": 2048,
  "batch_32": 128,
  "batch_256": 64,
  "gradient_accumulation_steps": 1,
  "learning_rate": 0.0002,
  "learning_rate_32_fine": 0.0002,
  "learning_rate_256_fine": 6e-05,
  "learning_rate_32_scratch": 0.0002,
  "learning_rate_256_scratch": 2e-05,
  "lr_warmup_steps": 500,
  "save_image_epochs": 20,
  "save_model_epochs": 5,
  "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",
  "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.9_SM_STOP_SIGN-BOX_new-set-1"
}