From a4d139033a21eebdb0484c5c4842281787babcea Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Fri, 15 Oct 2021 10:37:05 +0000 Subject: [PATCH] fix the bug of export model --- deepspeech/models/ds2/deepspeech2.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/deepspeech/models/ds2/deepspeech2.py b/deepspeech/models/ds2/deepspeech2.py index a2aa31f7f..63327a8c8 100644 --- a/deepspeech/models/ds2/deepspeech2.py +++ b/deepspeech/models/ds2/deepspeech2.py @@ -219,10 +219,10 @@ class DeepSpeech2Model(nn.Layer): The model built from pretrained result. """ model = cls( - #feat_size=dataloader.collate_fn.feature_size, - feat_size=dataloader.dataset.feature_size, - #dict_size=dataloader.collate_fn.vocab_size, - dict_size=dataloader.dataset.vocab_size, + feat_size=dataloader.collate_fn.feature_size, + #feat_size=dataloader.dataset.feature_size, + dict_size=dataloader.collate_fn.vocab_size, + #dict_size=dataloader.dataset.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,