folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

simple logger

Browse Source
This commit is contained in:
Dario Coscia
2023-07-14 12:04:29 +02:00
committed by Nicola Demo
parent 0ea17d8ff4
commit 955f904b05
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pina/solvers/pinn.py
View File

@@ -131,6 +131,10 @@ class PINN(SolverInterface):
# TODO Fix the bug, tot_loss is a label tensor without labels # 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 # we need to pass it as a torch tensor to make everything work
total_loss = sum(condition_losses) 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 return total_loss
@property @property