polished mle_losses

asyml · Apr 27, 2019 · dda394e · dda394e
1 parent b56d4f4
commit dda394e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/texar/losses/mle_losses.py b/texar/losses/mle_losses.py
@@ -35,6 +35,7 @@
     "binary_sigmoid_cross_entropy_with_clas"
 ]
 
+
 def sequence_softmax_cross_entropy(labels,
                                    logits,
                                    sequence_length,
@@ -116,6 +117,7 @@ def sequence_softmax_cross_entropy(labels,
 
         return losses
 
+
 def sequence_sparse_softmax_cross_entropy(labels,
                                           logits,
                                           sequence_length,
@@ -205,6 +207,7 @@ def sequence_sparse_softmax_cross_entropy(labels,
 
         return losses
 
+
 def sequence_sigmoid_cross_entropy(labels,
                                    logits,
                                    sequence_length,
@@ -303,6 +306,7 @@ class dimension. Must not set `average_across_classes`
 
         return losses
 
+
 def binary_sigmoid_cross_entropy(pos_logits=None,
                                  neg_logits=None,
                                  average_across_batch=True,
@@ -381,6 +385,7 @@ class dimension. Must not set `average_across_classes`
     else:
         return loss
 
+
 def binary_sigmoid_cross_entropy_with_clas(clas_fn,
                                            pos_inputs=None,
                                            neg_inputs=None,