From 786086486e8cb58c21554eb9bcb5890a13ad9c43 Mon Sep 17 00:00:00 2001 From: Junkun Date: Sun, 12 Sep 2021 21:25:26 -0700 Subject: [PATCH] fix norm_by_batchsize --- deepspeech/modules/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeech/modules/loss.py b/deepspeech/modules/loss.py index 023a19232..2c58be7e3 100644 --- a/deepspeech/modules/loss.py +++ b/deepspeech/modules/loss.py @@ -46,7 +46,7 @@ class CTCLoss(nn.Layer): if grad_norm_type == 'instance': self.norm_by_times = True if grad_norm_type == 'batch': - self.norm_by_times = True + self.norm_by_batchsize = True if grad_norm_type == 'frame': self.norm_by_total_logits_len = True