from local

This commit is contained in:
dohe0342 2022-12-12 14:26:58 +09:00
parent fa3a6a15f4
commit 8f15ad082a
3 changed files with 1 additions and 1 deletions

View File

@ -369,7 +369,7 @@ def decode_one_batch(
"""
device = next(model.parameters()).device
feature = batch["inputs"]
assert feature.ndim == 3
assert feature.ndim == 2 or feature.ndim == 3
feature = feature.to(device)
# at entry, feature is (N, T, C)