Log loss for debugging

w4ffl35 · Mar 12, 2023 · 76ea9c8 · 76ea9c8
1 parent 1a0c13c
commit 76ea9c8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py
@@ -1028,6 +1028,7 @@ def __call__(
 
                 # call the callback, if provided
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
+                    progress_bar.set_postfix({"loss": loss.loss.detach().item()})
                     progress_bar.update()
 
         # 11. Post-process the latents.