From 2e2fe93458e4671d758ab7888c9c9a545b2f3671 Mon Sep 17 00:00:00 2001
From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
Date: Fri, 6 Oct 2023 15:54:10 +0200
Subject: [PATCH] GAROM solver loss update

Changing from `LpLoss` to `PowerLoss`
---
 pina/solvers/garom.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/pina/solvers/garom.py b/pina/solvers/garom.py
index 7245e50..cfb9665 100644
--- a/pina/solvers/garom.py
+++ b/pina/solvers/garom.py
@@ -1,4 +1,5 @@
-""" Module for PINN """
+""" Module for GAROM """
+
 import torch
 try:
     from torch.optim.lr_scheduler import LRScheduler  # torch >= 2.0
@@ -8,7 +9,7 @@ except ImportError:
 from torch.optim.lr_scheduler import ConstantLR
 from .solver import SolverInterface
 from ..utils import check_consistency
-from ..loss import LossInterface, LpLoss
+from ..loss import LossInterface, PowerLoss
 from torch.nn.modules.loss import _Loss
 
 
@@ -58,7 +59,7 @@ class GAROM(SolverInterface):
             extra features for each.
         :param torch.nn.Module loss: The loss function used as minimizer,
             default ``None``. If ``loss`` is ``None`` the defualt
-            ``LpLoss(p=1)`` is used, as in the original paper.
+            ``PowerLoss(p=1)`` is used, as in the original paper.
         :param torch.optim.Optimizer optimizer_generator: The neural
             network optimizer to use for the generator network
             , default is `torch.optim.Adam`.
@@ -102,7 +103,7 @@ class GAROM(SolverInterface):
 
         # set loss
         if loss is None:
-            loss = LpLoss(p=1)
+            loss = PowerLoss(p=1)
         
         # check consistency 
         check_consistency(scheduler_generator, LRScheduler, subclass=True)
@@ -264,4 +265,4 @@ class GAROM(SolverInterface):
     
     @property
     def scheduler_discriminator(self):
-        return self._schedulers[1]
\ No newline at end of file
+        return self._schedulers[1]