From 06a410f90714d36134c0d6582fb6da2076a27231 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Thu, 14 Oct 2021 06:29:50 +0000 Subject: [PATCH] fix ctc_loss call for 2.2.0 --- deepspeech/__init__.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/deepspeech/__init__.py b/deepspeech/__init__.py index 493f10a6..90ab2223 100644 --- a/deepspeech/__init__.py +++ b/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, - input_lengths, label_lengths) + loss_out = paddle.fluid.layers.warpctc( + 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']