from local

This commit is contained in:
dohe0342 2023-06-09 16:52:30 +09:00
parent 1a88cc9dde
commit b4f5f9f26c
2 changed files with 2 additions and 0 deletions

View File

@ -849,6 +849,8 @@ def compute_loss(
warm_step = params.warm_step warm_step = params.warm_step
texts = batch["supervisions"]["text"] texts = batch["supervisions"]["text"]
print(texts)
exit()
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)