From 85674d22d4f994eaed463ae602c597bf2179fa30 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Fri, 10 Sep 2021 04:45:54 +0000
Subject: [PATCH] fix bugs

---
 deepspeech/models/u2.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/deepspeech/models/u2.py b/deepspeech/models/u2.py
index 46d932c98..fd8f15471 100644
--- 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__()