logging:
  filename: logging.log
  format: ' %(asctime)s - %(levelname)s -%(filename)s - %(funcName)s >> %(message)s'
  console_level: 20
  file_level: 10
wandb:
  project: slurpp
  reinit: true
model:
  name: slurpp_pipeline
  pretrained_path: stable-diffusion-2
  latent_scale_factor: 0.18215
  pretrained_path2: marigold-depth-v1-0
base_config:
- ../config/default.yaml
pipeline:
  name: SlurppPipeline
job_name: slurpp_release
dataloader:
  num_workers: 2
  effective_batch_size: 1
  max_train_batch_size: 1
  seed: 2024
  val_size: 20
  train_vis_size: 20
  image_size: 512
trainer:
  name: SlurppTrainer
  training_noise_scheduler:
    pretrained_path: stable-diffusion-2
  init_seed: 2024
  val_vis_frequency: 1
  num_hour_between_val: 2
  color_space: srgb
  inputs:
  - u
  inputs2:
  - u
  output:
  - clear
  output2:
  - bc
  - ill
  upsample: 1
  num_hour_between_real_vis: 2
max_epoch: 10000
max_iter: 100000
optimizer:
  name: Adam
loss:
  name: mse_loss
  kwargs:
    reduction: mean
lr: 1.0e-05
lr_scheduler:
  name: IterExponential
  kwargs:
    total_iter: 50000
    final_ratio: 0.01
    warmup_steps: 100
validation:
  denoising_steps: 50
  main_val_metric: psnr
  main_val_metric_goal: minimize
  init_seed: 2024
eval:
  eval_metrics:
  - psnr
  - ssim
  - lpips_loss
dual: true
rgb_loss: true
reconstruction_loss: true
one_step: true