File size: 1,357 Bytes
cc0e38f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
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
|