From 064d42137c744f24ba3c7dc37de5fee595231f13 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 023a1923..2c58be7e 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