from local

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

View File

@ -852,6 +852,7 @@ def compute_loss(
token_ids = sp.encode(texts, out_type=int)
y = k2.RaggedTensor(token_ids).to(device)
print(y)
with torch.set_grad_enabled(is_training):
simple_loss, pruned_loss, ctc_output = model(