提交 31922865 编写于 作者: H huangyuxin

fix some small mistakes

上级 61d85404
...@@ -139,9 +139,10 @@ class DeepSpeech2Trainer(Trainer): ...@@ -139,9 +139,10 @@ class DeepSpeech2Trainer(Trainer):
dict_size=self.train_loader.collate_fn.vocab_size, dict_size=self.train_loader.collate_fn.vocab_size,
num_conv_layers=config.model.num_conv_layers, num_conv_layers=config.model.num_conv_layers,
num_rnn_layers=config.model.num_rnn_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, num_fc_layers=config.model.num_fc_layers,
fc_layers_size_list=config.model.fc_layers_size_list, fc_layers_size_list=config.model.fc_layers_size_list,
rnn_size=config.model.rnn_layer_size,
use_gru=config.model.use_gru) use_gru=config.model.use_gru)
else: else:
raise Exception("wrong model type") raise Exception("wrong model type")
...@@ -411,9 +412,10 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): ...@@ -411,9 +412,10 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):
dict_size=self.test_loader.collate_fn.vocab_size, dict_size=self.test_loader.collate_fn.vocab_size,
num_conv_layers=config.model.num_conv_layers, num_conv_layers=config.model.num_conv_layers,
num_rnn_layers=config.model.num_rnn_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, num_fc_layers=config.model.num_fc_layers,
fc_layers_size_list=config.model.fc_layers_size_list, fc_layers_size_list=config.model.fc_layers_size_list,
rnn_size=config.model.rnn_layer_size,
use_gru=config.model.use_gru) use_gru=config.model.use_gru)
else: else:
raise Exception("Wrong model type") raise Exception("Wrong model type")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册