mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-12 14:27:26 +09:00
parent 8f15ad082a
commit 201efb32d7
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
View File

Binary file not shown.

11
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
View File

@ -375,7 +375,16 @@ def decode_one_batch(
# at entry, feature is (N, T, C) # at entry, feature is (N, T, C)
supervisions = batch["supervisions"] supervisions = batch["supervisions"]
feature_lens = supervisions["num_frames"].to(device) #feature_lens = supervisions["num_frames"].to(device)
if feature.ndim == 2:
feature_lens = []
for supervision in supervisions['cut']:
try: feature_lens.append(supervision.tracks[0].cut.recording.num_samples)
except: feature_lens.append(supervision.recording.num_samples)
feature_lens = torch.tensor(feature_lens)
elif feature.ndim == 3:
feature_lens = supervisions["num_frames"].to(device)
if params.simulate_streaming: if params.simulate_streaming:
feature_lens += params.left_context feature_lens += params.left_context