|
{ |
|
"project": "Common", |
|
"batch": 128, |
|
"epoch": 50, |
|
"learning_rate": 0.0002, |
|
"clean_rate": 1.0, |
|
"poison_rate": 0.5, |
|
"trigger": "NONE", |
|
"target": "FASHION", |
|
"dataset_load_mode": "FIXED", |
|
"gpu": "3", |
|
"ckpt": "res_DDPM-CIFAR10-32_CIFAR10_ep50_c1.0_p0.5_NONE-FASHION_new-set", |
|
"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": "sampling", |
|
"dataset": "CIFAR10", |
|
"output_dir": "res_DDPM-CIFAR10-32_CIFAR10_ep50_c1.0_p0.5_NONE-FASHION_new-set", |
|
"device_ids": [ |
|
0 |
|
], |
|
"clip": true |
|
} |