Backend paddle: convert loss_weights as same dtype of stacked losses (#…

…1780)
lululxvi · Jun 20, 2024 · e586a7b · e586a7b
1 parent ad6399b
commit e586a7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepxde/model.py b/deepxde/model.py
@@ -473,7 +473,7 @@ def outputs_losses(training, inputs, targets, auxiliary_vars, losses_fn):
             losses = paddle.stack(losses, axis=0)
             # Weighted losses
             if self.loss_weights is not None:
-                losses *= paddle.to_tensor(self.loss_weights)
+                losses *= paddle.to_tensor(self.loss_weights, dtype=losses.dtype)
             # Clear cached Jacobians and Hessians.
             grad.clear()
             return outputs_, losses