experiment: project_name: relighting_indoors output_dir: train_ex6_11 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 light_encoder: cross_dim: 768 K: 4 Bi: 4 Bc: 4 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 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: 40 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: 5 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