Update RBAPINN

Author: dario-coscia

pina/solvers/pinns/rbapinn.py

@@ -117,7 +117,7 @@ def __init__(
         # initialize weights
         self.weights = {}
         for condition_name in problem.conditions:
-            self.weights[condition_name] = 0
+            self.weights[condition_name] = 1
 
         # define vectorial loss
         self._vectorial_loss = deepcopy(loss)
@@ -158,7 +158,7 @@ def loss_phys(self, samples, equation):
         residual = self.compute_residual(samples=samples, equation=equation)
         cond = self.current_condition_name
 
-        r_norm = self.eta * torch.abs(residual) / torch.max(torch.abs(residual))
+        r_norm = self.eta * torch.abs(residual) / (torch.max(torch.abs(residual))+1e-12)
         self.weights[cond] = (self.gamma * self.weights[cond] + r_norm).detach()
 
         loss_value = self._vectorial_loss(