From 2683940ff4287fa158fb73d249e69c9e2715d6da Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 6 Oct 2022 16:11:50 +0530 Subject: [PATCH] control mlflow logging --- enhancer/models/model.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/enhancer/models/model.py b/enhancer/models/model.py index 6e6b4e1..cbbfad8 100644 --- a/enhancer/models/model.py +++ b/enhancer/models/model.py @@ -120,7 +120,11 @@ class Model(pl.LightningModule): loss = self.loss(prediction, target) - if self.logger: + if ( + (self.logger) + and (self.global_step > 50) + and (self.global_step % 50 == 0) + ): self.logger.experiment.log_metric( run_id=self.logger.run_id, key="train_loss", @@ -141,7 +145,11 @@ class Model(pl.LightningModule): self.log("val_metric", metric_val.item()) self.log("val_loss", loss_val.item()) - if self.logger: + if ( + (self.logger) + and (self.global_step > 50) + and (self.global_step % 50 == 0) + ): self.logger.experiment.log_metric( run_id=self.logger.run_id, key="val_loss",