refactor code of BC loss

lululxvi · Dec 18, 2024 · 85c8d2e · 85c8d2e
1 parent 292d05e
commit 85c8d2e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py
@@ -294,10 +294,8 @@ def forward_call(trunk_input):
                     end,
                     aux_var=model.net.auxiliary_vars[i][:, None],
                 )
-                error.append(error_i)
-            error = bkd.stack(error, 0)
-            loss = loss_fn(bkd.zeros_like(error), error)
-            losses.append(loss)
+                error.append(loss_fn(bkd.zeros_like(error_i), error_i))
+            losses.append(bkd.reduce_mean(bkd.stack(error, 0)))
         return losses
 
     def losses_train(self, targets, outputs, loss_fn, inputs, model, aux=None):