fix norm_by_batchsize

3 years ago · 064d42137c
parent 91bc5959a9
commit 064d42137c
1 changed files with 1 additions and 1 deletions
--- 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