from local

This commit is contained in:
dohe0342 2023-06-09 16:59:08 +09:00
parent fa8ec4171e
commit be5f456790
2 changed files with 1 additions and 2 deletions

View File

@ -832,9 +832,8 @@ def compute_loss(
# at entry, feature is (N, T, C)
assert feature.ndim == 2 or feature.ndim == 3
feature = feature.to(device)
supervisions = batch["supervisions"]
if feature.ndim == 2:
feature_lens = []
for supervision in supervisions['cut']: