FIx decode_stream.py

This commit is contained in:
Yifan Yang 2023-08-08 22:11:30 +08:00 committed by GitHub
parent 1ee251c8b3
commit 23e2c0fa96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -79,7 +79,7 @@ class DecodeStream(object):
self.pad_length = 7 + 2 * 3
if params.decoding_method == "greedy_search":
self.hyp = [params.blank_id] * params.context_size
self.hyp = [-1] * (params.context_size - 1) + [params.blank_id]
elif params.decoding_method == "modified_beam_search":
self.hyps = HypothesisList()
self.hyps.add(