diff --git a/tests/unit/asr/u2_model_test.py b/tests/unit/asr/u2_model_test.py index 5b11d2ad3e2d42e7bd6d291dcc633475ffd034c2..f46c6d403ed3855a42286297dac828e2bd7fb2da 100644 --- 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)