fix the bug of export model

pull/899/head
huangyuxin 3 years ago
parent f47058fa3f
commit 21d6c1cee2

@ -219,10 +219,10 @@ class DeepSpeech2Model(nn.Layer):
The model built from pretrained result. The model built from pretrained result.
""" """
model = cls( model = cls(
#feat_size=dataloader.collate_fn.feature_size, feat_size=dataloader.collate_fn.feature_size,
feat_size=dataloader.dataset.feature_size, #feat_size=dataloader.dataset.feature_size,
#dict_size=dataloader.collate_fn.vocab_size, dict_size=dataloader.collate_fn.vocab_size,
dict_size=dataloader.dataset.vocab_size, #dict_size=dataloader.dataset.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_size=config.model.rnn_layer_size,

Loading…
Cancel
Save