from local

This commit is contained in:
dohe0342 2023-01-25 16:42:49 +09:00
parent 19b1733b61
commit 512e905c2f
3 changed files with 2 additions and 1 deletions

View File

@ -832,7 +832,8 @@ def compute_loss(
batch_idx_train = params.batch_idx_train
warm_step = params.warm_step
texts = batch["supervisions"]["text"]
#texts = batch["supervisions"]["text"]
texts = batch["supervisions"]["greedy pseudo text"]
token_ids = sp.encode(texts, out_type=int)
y = k2.RaggedTensor(token_ids).to(device)