From 061d11d55df354c77686f71c193aa07e14186632 Mon Sep 17 00:00:00 2001 From: Tim Date: Fri, 30 Jun 2023 13:29:02 -0700 Subject: [PATCH] Fixing validation step for PL 1 --- main.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/main.py b/main.py index 66b74d1..c916f51 100644 --- a/main.py +++ b/main.py @@ -469,9 +469,8 @@ class ImageLogger(Callback): self.log_img(pl_module, batch, batch_idx, split="train") @rank_zero_only - # def on_validation_batch_end(self, trainer, pl_module, outputs, batch, batch_idx): def on_validation_batch_end( - self, trainer, pl_module, outputs, batch, batch_idx, **kwargs + self, trainer, pl_module, outputs, batch, batch_idx, *args, **kwargs ): if not self.disabled and pl_module.global_step > 0: self.log_img(pl_module, batch, batch_idx, split="val")