Merge pull request #892 from PaddlePaddle/ctc_grad_norm

[WIP] ctc loss api debug

deepspeech/__init__.py

@@ -362,11 +362,19 @@ def ctc_loss(logits,
              label_lengths,
              blank=0,
              reduction='mean',
-             norm_by_times=True):
+             norm_by_times=True,
+             norm_by_batchsize=False,
+             norm_by_total_logits_len=False):
     #logger.info("my ctc loss with norm by times")
     ## https://github.com/PaddlePaddle/Paddle/blob/f5ca2db2cc/paddle/fluid/operators/warpctc_op.h#L403
-    loss_out = paddle.fluid.layers.warpctc(logits, labels, blank, norm_by_times,
+    loss_out = paddle.fluid.layers.warpctc(
-                                           input_lengths, label_lengths)
+        logits,
+        labels,
+        blank,
+        norm_by_times,
+        input_lengths,
+        label_lengths,
+        norm_by_batchsize, )
 
     loss_out = paddle.fluid.layers.squeeze(loss_out, [-1])
     assert reduction in ['mean', 'sum', 'none']