from local

This commit is contained in:
dohe0342 2023-06-09 16:53:31 +09:00
parent c16039addb
commit 7fc02a434a
2 changed files with 1 additions and 0 deletions

View File

@ -850,6 +850,7 @@ def compute_loss(
texts = batch["supervisions"]["text"] texts = batch["supervisions"]["text"]
texts = [text.upper() for text in texts] texts = [text.upper() for text in texts]
print(texts)
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)