| { | |
| "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" | |
| } |