From 68f3f89bd3e2a7e13df48a170f65ae768fabfdc6 Mon Sep 17 00:00:00 2001 From: ablattmann Date: Wed, 26 Jul 2023 10:13:42 +0200 Subject: [PATCH] Fix crashing line in logging in `sgm/models/diffusion.py` (#64) --- sgm/models/diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sgm/models/diffusion.py b/sgm/models/diffusion.py index 8b7cc17..3ffb146 100644 --- a/sgm/models/diffusion.py +++ b/sgm/models/diffusion.py @@ -307,7 +307,7 @@ class DiffusionEngine(pl.LightningModule): log["inputs"] = x z = self.encode_first_stage(x) log["reconstructions"] = self.decode_first_stage(z) - logger.update(self.log_conditionings(batch, N)) + log.update(self.log_conditionings(batch, N)) for k in c: if isinstance(c[k], torch.Tensor):