From 441b7484348953deb7c94150675d54583ef5a81a Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 10 Aug 2021 23:25:13 +0200 Subject: Update to config and logging in VQGAN --- text_recognizer/models/vqgan.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) (limited to 'text_recognizer/models/vqgan.py') diff --git a/text_recognizer/models/vqgan.py b/text_recognizer/models/vqgan.py index 80653b6..7c707b1 100644 --- a/text_recognizer/models/vqgan.py +++ b/text_recognizer/models/vqgan.py @@ -39,11 +39,8 @@ class VQGANLitModel(BaseLitModel): "train/loss", loss, prog_bar=True, - logger=True, - on_step=True, - on_epoch=True, ) - self.log_dict(log, prog_bar=False, logger=True, on_step=True, on_epoch=True) + self.log_dict(log, logger=True, on_step=True, on_epoch=True) return loss if optimizer_idx == 1: @@ -58,11 +55,8 @@ class VQGANLitModel(BaseLitModel): "train/discriminator_loss", loss, prog_bar=True, - logger=True, - on_step=True, - on_epoch=True, ) - self.log_dict(log, prog_bar=False, logger=True, on_step=True, on_epoch=True) + self.log_dict(log, logger=True, on_step=True, on_epoch=True) return loss def validation_step(self, batch: Tuple[Tensor, Tensor], batch_idx: int) -> None: @@ -78,7 +72,7 @@ class VQGANLitModel(BaseLitModel): stage="val", ) self.log( - "val/loss", loss, prog_bar=True, logger=True, on_step=True, on_epoch=True + "val/loss", loss, prog_bar=True, ) self.log_dict(log) -- cgit v1.2.3-70-g09d2