softmax_cross_entropy supports TF and TF.compat.v1

lululxvi · lululxvi · commit d1a896d430b4 · 2021-07-02T00:43:47.000-04:00
diff --git a/deepxde/losses.py b/deepxde/losses.py
@@ -22,7 +22,7 @@ def zero(*_):
 
 mean_absolute_error = tf.keras.losses.MeanAbsoluteError()
 mean_absolute_percentage_error = tf.keras.losses.MeanAbsolutePercentageError()
-softmax_cross_entropy = tf.losses.softmax_cross_entropy
+softmax_cross_entropy = tf.keras.losses.CategoricalCrossentropy(from_logits=True)
 
 
 def get(identifier):
