| { | |
| "project": "Poison_Rates", | |
| "batch": 128, | |
| "epoch": 50, | |
| "learning_rate": 0.0002, | |
| "clean_rate": 1.0, | |
| "poison_rate": 0.1, | |
| "trigger": "SM_BOX_MED", | |
| "target": "TRIGGER", | |
| "dataset_load_mode": "FIXED", | |
| "gpu": "2", | |
| "ckpt": "DDPM-CIFAR10-32", | |
| "overwrite": true, | |
| "postfix": "new-set", | |
| "fclip": "w", | |
| "save_image_epochs": 20, | |
| "save_model_epochs": 5, | |
| "is_save_all_model_epochs": false, | |
| "sample_ep": null, | |
| "result": ".", | |
| "eval_sample_n": 16, | |
| "measure_sample_n": 2048, | |
| "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": "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", | |
| "ep_model_dir": "epochs", | |
| "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.1_SM_BOX_MED-TRIGGER_new-set" | |
| } |