Skip to content

Commit a31d197

Browse files
committed
Change error message
1 parent eacdc41 commit a31d197

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

deepxde/model.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,9 @@ def _compile_pytorch(self, lr, loss_fn, decay):
285285
elif self.net.regularizer[0] == "l2":
286286
l2_factor = self.net.regularizer[1]
287287
else:
288-
raise ValueError(f"Unknown regularizer name: {self.net.regularizer[0]}")
288+
raise NotImplementedError(
289+
f"{self.net.regularizer[0]} regularizer hasn't been implemented for backend pytorch."
290+
)
289291

290292
def outputs(training, inputs):
291293
self.net.train(mode=training)
@@ -322,7 +324,7 @@ def outputs_losses(training, inputs, targets, auxiliary_vars, losses_fn):
322324
losses = losses_fn(targets, outputs_, loss_fn, inputs, self, aux=aux)
323325
if not isinstance(losses, list):
324326
losses = [losses]
325-
if l1_factor:
327+
if l1_factor > 0:
326328
l1_loss = torch.sum(
327329
torch.stack([torch.sum(p.abs()) for p in self.net.parameters()])
328330
)

0 commit comments

Comments
 (0)