config.yaml

_wandb:
    value:
        cli_version: 0.19.1
        m: []
        python_version: 3.10.12
        t:
            "1":
                - 1
                - 2
                - 3
                - 11
                - 12
                - 41
                - 45
                - 49
                - 51
                - 55
                - 71
                - 83
            "2":
                - 1
                - 2
                - 3
                - 11
                - 12
                - 41
                - 45
                - 49
                - 51
                - 55
                - 71
                - 83
            "3":
                - 2
                - 23
                - 55
                - 61
            "4": 3.10.12
            "5": 0.19.1
            "6": 4.44.2
            "8":
                - 2
                - 5
            "12": 0.19.1
            "13": linux-x86_64
adam_beta1:
    value: 0.9
adam_beta2:
    value: 0.999
adam_epsilon:
    value: 1e-08
adam_weight_decay:
    value: 0.01
allow_tf32:
    value: false
cache_dir:
    value: null
caption_column:
    value: captions
checkpointing_steps:
    value: 10000
checkpoints_total_limit:
    value: null
conditioning_image_column:
    value: canny_edges
controlnet_model_name_or_path:
    value: null
dataloader_num_workers:
    value: 0
dataset_config_name:
    value: null
dataset_name:
    value: ButterChicken98/test-controlnet-full
enable_xformers_memory_efficient_attention:
    value: false
gradient_accumulation_steps:
    value: 1
gradient_checkpointing:
    value: false
hub_model_id:
    value: plantVillage-stableDiffusion-2-controlnet_iter1
hub_token:
    value: null
image_column:
    value: image
learning_rate:
    value: 1e-05
logging_dir:
    value: logs
lr_num_cycles:
    value: 1
lr_power:
    value: 1
lr_scheduler:
    value: constant
lr_warmup_steps:
    value: 500
max_grad_norm:
    value: 1
max_train_samples:
    value: null
max_train_steps:
    value: 15750
mixed_precision:
    value: null
num_train_epochs:
    value: 7
num_validation_images:
    value: 4
output_dir:
    value: /kaggle/working/
pretrained_model_name_or_path:
    value: stabilityai/stable-diffusion-2-1-base
proportion_empty_prompts:
    value: 0
push_to_hub:
    value: true
report_to:
    value: wandb
resolution:
    value: 256
resume_from_checkpoint:
    value: null
revision:
    value: null
scale_lr:
    value: false
seed:
    value: null
set_grads_to_none:
    value: false
tokenizer_name:
    value: null
tracker_project_name:
    value: controlnet
train_batch_size:
    value: 4
train_data_dir:
    value: null
use_8bit_adam:
    value: false
validation_steps:
    value: 1000
variant:
    value: null