config_files/losses.yaml

diffusion_losses:
- name: L2Loss
  weight: 1
lcm_losses:
- name: HuberLoss
  weight: 1
# - name: DINOLoss
#   weight: 1e-3
# - name: L2Loss
#   weight: 5e-2
# - name: LPIPSLoss
#   weight: 1e-3
# - name: DreamSIMLoss
#   weight: 1e-3
# - name: IDLoss
#   weight: 1e-3
#   visualize_every_k: 50
#   init_params:
#     pretrained_arcface_path: /home/dcor/orlichter/consistency_encoder_private/pretrained_models/model_ir_se50.pth