FIX: Use keras loss (#278)

seanpmorgan · web-flow · commit 66b5fca45c23 · 2019-06-11T11:29:31.000-04:00
diff --git a/tensorflow_addons/seq2seq/loss.py b/tensorflow_addons/seq2seq/loss.py
@@ -143,7 +143,7 @@ def sequence_loss(logits,
         return crossent
 
 
-class SequenceLoss(tf.losses.Loss):
+class SequenceLoss(tf.keras.losses.Loss):
     """Weighted cross-entropy loss for a sequence of logits."""
 
     def __init__(self,
