| adam_beta1: 0.9 |
| adam_beta2: 0.99 |
| adam_epsilon: 1.0e-08 |
| adam_weight_decay: 0.01 |
| adam_weight_decay_text_encoder: null |
| allow_tf32: false |
| blur_kernel_size_range: 21@51 |
| budget: 1 |
| censor_modes: pixelation@gaussian_blur@black |
| center_crop: false |
| checkpointing_steps: 500 |
| dataloader_num_workers: 0 |
| dataset_name: timm/imagenet-1k-wds |
| gradient_accumulation_steps: 8 |
| gradient_checkpointing: false |
| image_column: jpg |
| image_log_steps: 50 |
| learning_rate: 1.0 |
| logging_dir: logs |
| max_grad_norm: 1.0 |
| max_train_samples: null |
| max_train_steps: 10000 |
| mixed_precision: null |
| num_train_epochs: 2 |
| output_dir: outputs |
| pixelation_block_size_range: 10@40 |
| pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5 |
| prodigy_beta3: null |
| prodigy_decouple: true |
| prodigy_safeguard_warmup: true |
| prodigy_use_bias_correction: true |
| random_flip: false |
| rank: 64 |
| report_to: tensorboard |
| resolution: 520 |
| resume_from_checkpoint: null |
| seed: null |
| train_batch_size: 2 |
|
|
