| 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: 2 | |
| censor_modes: pixelation@gaussian_blur@black | |
| center_crop: false | |
| checkpointing_steps: 1000 | |
| cover_image_column: null | |
| cover_image_mode: censorship | |
| dataloader_num_workers: 0 | |
| dataset_name: timm/imagenet-1k-wds | |
| disable_n_sq: false | |
| enable_n_crop: false | |
| enable_n_fq: false | |
| gradient_accumulation_steps: 4 | |
| gradient_checkpointing: false | |
| image_log_steps: 50 | |
| jpeg_qf: 90 | |
| learning_rate: 1.0 | |
| logging_dir: logs | |
| make_cover_image: true | |
| max_grad_norm: 1.0 | |
| max_train_samples: null | |
| max_train_steps: 10000 | |
| message_image_column: jpg | |
| mixed_precision: null | |
| num_train_epochs: 1 | |
| output_dir: /work/ernestchu0/censorship/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: 12345 | |
| train_batch_size: 2 | |
| train_data_size: null | |