Merge pull request #822 from LittleChenCc/develop

fix norm_by_batchsize
3 years ago · 7e2c6d4f7f
parent 91bc5959a9 064d42137c
commit 7e2c6d4f7f
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