From 955f904b05b4a0ed6098e2e085fdde8f9044131c Mon Sep 17 00:00:00 2001
From: Dario Coscia <dariocoscia@cli-10-110-4-217.WIFIeduroamSTUD.units.it>
Date: Fri, 14 Jul 2023 12:04:29 +0200
Subject: [PATCH] simple logger

---
 pina/solvers/pinn.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pina/solvers/pinn.py b/pina/solvers/pinn.py
index 4ec03a0..fd561fc 100644
--- a/pina/solvers/pinn.py
+++ b/pina/solvers/pinn.py
@@ -131,6 +131,10 @@ class PINN(SolverInterface):
         # TODO Fix the bug, tot_loss is a label tensor without labels
         # we need to pass it as a torch tensor to make everything work
         total_loss = sum(condition_losses)
+
+        self.log('mean_loss', float(total_loss / len(condition_losses)), prog_bar=True, logger=False)
+        for condition_loss, loss in zip(self.problem.conditions, condition_losses):
+            self.log(condition_loss + '_loss', float(loss), prog_bar=True, logger=False)
         return total_loss
 
     @property