fix asr unit test

4 years ago · 551e8b28eb
parent a6cd3c6b84
commit 551e8b28eb
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)