Method log_images

ldm/models/diffusion/ddpm.py:1802–1805

(self, *args, **kwargs)

    @torch.no_grad()
    def log_images(self, *args, **kwargs):
        log = super().log_images(*args, **kwargs)
        log["lr"] = rearrange(args[0]["lr"], 'b h w c -> b c h w')
        return log