From 319228653e9a360bccaf5cdbc1e3a413bc27a3d9 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Thu, 5 Aug 2021 05:27:45 +0000 Subject: [PATCH] fix some small mistakes --- deepspeech/exps/deepspeech2/model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/deepspeech/exps/deepspeech2/model.py b/deepspeech/exps/deepspeech2/model.py index de5ff5f4..03974e1b 100644 --- a/deepspeech/exps/deepspeech2/model.py +++ b/deepspeech/exps/deepspeech2/model.py @@ -139,9 +139,10 @@ class DeepSpeech2Trainer(Trainer): dict_size=self.train_loader.collate_fn.vocab_size, num_conv_layers=config.model.num_conv_layers, num_rnn_layers=config.model.num_rnn_layers, + rnn_size=config.model.rnn_layer_size, + rnn_direction=config.model.rnn_direction, num_fc_layers=config.model.num_fc_layers, fc_layers_size_list=config.model.fc_layers_size_list, - rnn_size=config.model.rnn_layer_size, use_gru=config.model.use_gru) else: raise Exception("wrong model type") @@ -411,9 +412,10 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): dict_size=self.test_loader.collate_fn.vocab_size, num_conv_layers=config.model.num_conv_layers, num_rnn_layers=config.model.num_rnn_layers, + rnn_size=config.model.rnn_layer_size, + rnn_direction=config.model.rnn_direction, num_fc_layers=config.model.num_fc_layers, fc_layers_size_list=config.model.fc_layers_size_list, - rnn_size=config.model.rnn_layer_size, use_gru=config.model.use_gru) else: raise Exception("Wrong model type")