| batch: 128 | |
| batch_256: 64 | |
| batch_32: 128 | |
| ckpt: NCSN_CIFAR10_my | |
| ckpt_dir: ckpt | |
| ckpt_path: NCSN_CIFAR10_my/ckpt | |
| clean_rate: 1.0 | |
| clip: false | |
| data_ckpt_dir: data.ckpt | |
| data_ckpt_path: NCSN_CIFAR10_my/data.ckpt | |
| dataset: CIFAR10 | |
| dataset_load_mode: FIXED | |
| dataset_path: datasets | |
| ep_model_dir: epochs | |
| epoch: 800 | |
| eval_max_batch: 1500 | |
| eval_sample_n: 16 | |
| fclip: o | |
| gpu: '0' | |
| gradient_accumulation_steps: 1 | |
| hub_private_repo: false | |
| infer_steps: 1000 | |
| is_save_all_model_epochs: false | |
| learning_rate: 0.0001 | |
| learning_rate_256_scratch: 2.0e-05 | |
| learning_rate_32_scratch: 0.0002 | |
| lr_warmup_steps: 500 | |
| measure_sample_n: 10000 | |
| mixed_precision: 'no' | |
| mode: sampling | |
| output_dir: NCSN_CIFAR10_my | |
| overwrite: true | |
| overwrite_output_dir: true | |
| poison_rate: 0.0 | |
| postfix: ve_from_scratch | |
| project: Default | |
| psi: 0.0 | |
| push_to_hub: false | |
| result: res_ve_from_scratch | |
| save_image_epochs: 20 | |
| save_model_epochs: 1 | |
| sde_type: SDE-VE | |
| seed: 0 | |
| solver_type: sde | |
| target: TRIGGER | |
| task: generate | |
| trigger: NONE | |
| ve_scale: 1.0 | |
| vp_scale: 1.0 | |