wandb/run-20230711_234436-fn1ccjrl/files/config.yaml

wandb_version: 1

_wandb:
  desc: null
  value:
    python_version: 3.8.10
    cli_version: 0.15.4
    framework: huggingface
    huggingface_version: 4.30.2
    is_jupyter_run: false
    is_kaggle_kernel: true
    start_time: 1689133476.354589
    t:
      1:
      - 1
      - 11
      - 41
      - 49
      - 51
      - 55
      - 71
      - 83
      2:
      - 1
      - 11
      - 41
      - 49
      - 51
      - 55
      - 71
      - 83
      3:
      - 23
      - 24
      4: 3.8.10
      5: 0.15.4
      6: 4.30.2
      8:
      - 2
      - 5
pretrained_model_name_or_path:
  desc: null
  value: runwayml/stable-diffusion-v1-5
revision:
  desc: null
  value: null
dataset_name:
  desc: null
  value: null
dataset_config_name:
  desc: null
  value: null
train_data_dir:
  desc: null
  value: /l/vision/v5/sragas/easel_ai/thumbs_up_srimanth_plain/
image_column:
  desc: null
  value: image
caption_column:
  desc: null
  value: text
validation_prompt:
  desc: null
  value: srimanth thumbs up
num_validation_images:
  desc: null
  value: 4
validation_epochs:
  desc: null
  value: 1
max_train_samples:
  desc: null
  value: null
output_dir:
  desc: null
  value: /l/vision/v5/sragas/easel_ai/models_srimanth_plain/
cache_dir:
  desc: null
  value: /l/vision/v5/sragas/hf_models/
seed:
  desc: null
  value: 15
resolution:
  desc: null
  value: 512
center_crop:
  desc: null
  value: true
random_flip:
  desc: null
  value: true
train_batch_size:
  desc: null
  value: 2
num_train_epochs:
  desc: null
  value: 300
max_train_steps:
  desc: null
  value: 1800
gradient_accumulation_steps:
  desc: null
  value: 4
gradient_checkpointing:
  desc: null
  value: false
learning_rate:
  desc: null
  value: 1.0e-05
scale_lr:
  desc: null
  value: false
lr_scheduler:
  desc: null
  value: cosine
lr_warmup_steps:
  desc: null
  value: 500
snr_gamma:
  desc: null
  value: null
use_8bit_adam:
  desc: null
  value: false
allow_tf32:
  desc: null
  value: false
dataloader_num_workers:
  desc: null
  value: 0
adam_beta1:
  desc: null
  value: 0.9
adam_beta2:
  desc: null
  value: 0.999
adam_weight_decay:
  desc: null
  value: 0.01
adam_epsilon:
  desc: null
  value: 1.0e-08
max_grad_norm:
  desc: null
  value: 1.0
push_to_hub:
  desc: null
  value: true
hub_token:
  desc: null
  value: null
prediction_type:
  desc: null
  value: null
hub_model_id:
  desc: null
  value: srimanth-thumbs-up-lora-plain
logging_dir:
  desc: null
  value: logs
mixed_precision:
  desc: null
  value: null
report_to:
  desc: null
  value: wandb
local_rank:
  desc: null
  value: 0
checkpointing_steps:
  desc: null
  value: 500
checkpoints_total_limit:
  desc: null
  value: null
resume_from_checkpoint:
  desc: null
  value: null
enable_xformers_memory_efficient_attention:
  desc: null
  value: false
noise_offset:
  desc: null
  value: 0
rank:
  desc: null
  value: 4