mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-08 08:34:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

check codebook index range before saving

Browse Source
This commit is contained in:
Guo Liyong 2022-04-29 11:57:51 +08:00
parent 76e56fa28f
commit c026a97d41
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/vq_pruned_transducer_stateless2/hubert_code_indices.py
View File

@ -92,9 +92,7 @@ def compute_codeindices(
# [N, T, C] # [N, T, C]
codebook_indices = codebook_indices.to("cpu").numpy() codebook_indices = codebook_indices.to("cpu").numpy()
assert np.all( assert np.min(codebook_indices) >= 0
codebook_indices[np.where(codebook_indices < 0)] == -100
)
assert np.max(codebook_indices) < 256 assert np.max(codebook_indices) < 256
supervisions = batch["supervisions"] supervisions = batch["supervisions"]