File size: 1,128 Bytes
c1318a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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 |