From 80b4b43460d7260bce0315159ef5f6cc4284c597 Mon Sep 17 00:00:00 2001 From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> Date: Fri, 6 Oct 2023 15:47:06 +0200 Subject: [PATCH] GAROM loggers (#181) * adding loggers for GAROM solver --------- Co-authored-by: Dario Coscia --- pina/solvers/garom.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/pina/solvers/garom.py b/pina/solvers/garom.py index e0ac376..7245e50 100644 --- a/pina/solvers/garom.py +++ b/pina/solvers/garom.py @@ -231,6 +231,12 @@ class GAROM(SolverInterface): self.k += self.lambda_k * diff.item() self.k = min(max(self.k, 0), 1) # Constraint to interval [0, 1] + # logging + self.log('mean_loss', float(r_loss), prog_bar=True, logger=True) + self.log('d_loss', float(d_loss), prog_bar=True, logger=True) + self.log('g_loss', float(g_loss), prog_bar=True, logger=True) + self.log('stability_metric', float(d_loss_real + torch.abs(diff)), prog_bar=True, logger=True) + else: raise NotImplementedError('GAROM works only in data-driven mode.')