| 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 | |