Skip to content

Commit c1a33fe

Browse files
committed
Bug fix: MFNN has nan in test loss
1 parent 303ae80 commit c1a33fe

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

deepxde/data/mf.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,9 @@ def __init__(
109109

110110
def losses(self, targets, outputs, loss, model):
111111
n = tf.cond(model.net.training, lambda: len(self.X_lo_train), lambda: 0)
112-
loss_lo = loss(targets[0][:n], outputs[0][:n])
112+
loss_lo = tf.cond(
113+
n > 0, lambda: loss(targets[0][:n], outputs[0][:n]), lambda: 0.0
114+
)
113115
loss_hi = loss(targets[1][n:], outputs[1][n:])
114116
return [loss_lo, loss_hi]
115117

0 commit comments

Comments
 (0)