From 8fef667f801d58d482f32a3bb6724d1157158c74 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Fri, 10 Sep 2021 04:34:31 +0000 Subject: [PATCH] fix u2 ctc config bugs --- deepspeech/models/u2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeech/models/u2.py b/deepspeech/models/u2.py index 7061d173..46d932c9 100644 --- a/deepspeech/models/u2.py +++ b/deepspeech/models/u2.py @@ -867,7 +867,7 @@ class U2Model(U2BaseModel): odim=vocab_size, enc_n_units=encoder.output_size(), blank_id=0, - dropout_rate=model_conf['ctc_dropout_rate'], + dropout_rate=model_conf['ctc_dropoutrate'], reduction=True, # sum batch_average=True, # sum / batch_size grad_norm_type=model_conf['ctc_grad_norm_type'])