Skip to content

Commit 3a06438

Browse files
remove target_indices with len 0
1 parent 0775277 commit 3a06438

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

users/schmitt/experiments/exp2026_04_09_unsupervised_asr/models/train_steps/aed_denoising_discrete.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,8 @@ def train_step(
9393
target_indices_: ReturnnTensor = extern_data["target"]
9494
target_indices: Tensor = target_indices_.raw_tensor
9595
target_indices_lens: Tensor = target_indices_.dims[1].dyn_size_ext.raw_tensor
96+
target_indices = target_indices[target_indices_lens > 0]
97+
target_indices_lens = target_indices_lens[target_indices_lens > 0]
9698
else:
9799
target_indices_ = label_indices_
98100
target_indices = label_indices

0 commit comments

Comments
 (0)