experiment:
  project_name: relighting_indoors
  output_dir: train_ex8_8
  logging_dir: logs
  report_to: tensorboard
model:
  pretrained_model_name_or_path: /mnt/HDD3/miayan/paper/envs/huggingface_cache/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6
  revision: null
  variant: null
  tokenizer_name: null
  hf_repo_id: Miayan/project
  hf_version: null
  pretrain_unet_path: scribblelight_controlnet/checkpoint-10000
  resume_from_checkpoint: latest
  enable_ambient_cond: true
light_encoder:
  cross_dim: 768
  K: 4
  Bi: 4
  Bc: 4
  Ba: 4
albedo_estimator:
  enabled: true
  version: v2
  load_stage: 3
  frozen_components:
  - ord_model
  - iid_model
  - col_model
  trainable_components:
  - alb_model
data:
  dataset_cache_dir: /mnt/HDD3/miayan/paper/relighting_datasets/
  data_hf_repo_id: Miayan/physical-relighting-dataset
  data_split: test
  resolution: 256
  dataloader_num_workers: 0
  condition_mode: lightmap_normal
  relighting_impl: gemini_amb
  colors:
  - - 255
    - 255
    - 255
  - - 255
    - 0
    - 0
  - - 0
    - 255
    - 0
  - - 0
    - 0
    - 255
  - - 255
    - 255
    - 0
  - - 255
    - 165
    - 0
  - - 128
    - 0
    - 128
  - - 255
    - 192
    - 203
  - - 0
    - 255
    - 255
  - - 255
    - 0
    - 255
  intensities:
  - 0.0
  - 0.1
  - 0.2
  - 0.4
  - 0.7
  - 1.0
training:
  batch_size: 1
  num_epochs: 20
  max_train_steps: null
  max_train_samples: null
  seed: 42
  mixed_precision: 'no'
  allow_tf32: true
  gradient_accumulation_steps: 1
  checkpointing_steps: 5000
  checkpoints_total_limit: null
  learning_rate: 5.0e-06
  scale_lr: false
  lr_scheduler: constant
  lr_warmup_steps: 500
  adam_beta1: 0.9
  adam_beta2: 0.999
  adam_weight_decay: 0.01
  adam_epsilon: 1.0e-08
  max_grad_norm: 1.0
  use_8bit_adam: false
  set_grads_to_none: false
losses:
  phys_loss:
    enabled: true
    weight: 1.0
  latent_loss:
    enabled: true
    weight: 1.0
  image_loss:
    enabled: true
    weight: 1.0
  area_loss:
    enabled: false
    weight: 10.0
  intensity_loss:
    enabled: false
    weight: 1.0
  recon_loss:
    enabled: false
    weight: 1.0
    keep_aux_models_on_gpu: true
  sup_structure_loss:
    enabled: true
    weight: 1.0
    enable_decay: false
    start_weight: 1.0
    end_weight: 0.1
    decay_steps: 5000
  self_recon_loss:
    enabled: true
    weight: 2.0
  consistency_loss:
    enabled: true
    weight: 1.0