from local

This commit is contained in:
dohe0342 2023-06-09 17:00:29 +09:00
parent 5a3e26e1e5
commit 217b18e24f
2 changed files with 0 additions and 1 deletions

View File

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