fix asr unit test

3 years ago · d8e4b31405
parent e369022f71
commit d8e4b31405
1 changed files with 4 additions and 0 deletions
--- a/tests/unit/asr/u2_model_test.py
+++ b/tests/unit/asr/u2_model_test.py
@ -74,6 +74,8 @@ class TestU2Model(unittest.TestCase):
            model_conf:
                ctc_weight: 0.3
                lsm_weight: 0.1     # label smoothing option
+                ctc_dropoutrate: 0.0
+                ctc_grad_norm_type: null
                length_normalized_loss: false
        """
        cfg = CN().load_cfg(conf_str)
@ -126,6 +128,8 @@ class TestU2Model(unittest.TestCase):
            model_conf:
                ctc_weight: 0.3
                lsm_weight: 0.1     # label smoothing option
+                ctc_dropoutrate: 0.0
+                ctc_grad_norm_type: null
                length_normalized_loss: false
        """
        cfg = CN().load_cfg(conf_str)