adam_beta1: 0.9
adam_beta2: 0.999
adam_epsilon: 1e-08
adam_weight_decay: 0.01
allow_tf32: False
cache_dir: None
caption_column: text
center_crop: True
checkpointing_steps: 10000
checkpoints_total_limit: None
dataloader_num_workers: 0
dataset_config_name: None
dataset_name: None
enable_xformers_memory_efficient_attention: False
freeze_unet: False
gradient_accumulation_steps: 4
gradient_checkpointing: False
hub_model_id: None
hub_token: None
image_column: image
learning_rate: 1e-06
local_rank: 0
logging_dir: logs
lr_scheduler: cosine
lr_warmup_steps: 0
max_grad_norm: 1.0
max_train_samples: None
max_train_steps: None
mixed_precision: bf16
non_ema_revision: None
num_train_epochs: 1
output_dir: /home/data/ckpts/hbansal/sd_finetune/sup_1k
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
push_to_hub: False
random_flip: True
report_to: None
resolution: 512
resume_from_checkpoint: None
revision: None
scale_lr: False
seed: None
train_batch_size: 1
train_data_file: /home/data/datasets/ImageNet1K/ILSVRC/Data/CLS-LOC/sup_classname_imagepath.csv
unsupervised: False
use_8bit_adam: True
use_ema: False
wandb: True