fix bugs

4 years ago · 85674d22d4
parent f0470e0529
commit 85674d22d4
1 changed files with 2 additions and 1 deletions
--- a/deepspeech/models/u2.py
+++ b/deepspeech/models/u2.py
@ -115,7 +115,8 @@ class U2BaseModel(nn.Layer):
                 ctc_weight: float=0.5,
                 ignore_id: int=IGNORE_ID,
                 lsm_weight: float=0.0,
-                 length_normalized_loss: bool=False):
+                 length_normalized_loss: bool=False,
+                 **kwargs):
        assert 0.0 <= ctc_weight <= 1.0, ctc_weight

        super().__init__()