Fix aishell tdnn_lstm_ctc decoding (#149)

This commit is contained in:
Wei Kang 2021-12-14 14:42:58 +08:00 committed by GitHub
parent a183d5bfd7
commit 76a51bf037
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions

View File

@ -15,7 +15,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import argparse
import logging
from collections import defaultdict
@ -177,7 +176,7 @@ def decode_one_batch(
lattice = get_lattice(
nnet_output=nnet_output,
HLG=HLG,
decoding_graph=HLG,
supervision_segments=supervision_segments,
search_beam=params.search_beam,
output_beam=params.output_beam,

View File

@ -16,7 +16,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import argparse
import logging
import math
@ -191,7 +190,7 @@ def main():
lattice = get_lattice(
nnet_output=nnet_output,
HLG=HLG,
decoding_graph=HLG,
supervision_segments=supervision_segments,
search_beam=params.search_beam,
output_beam=params.output_beam,