from local

This commit is contained in:
dohe0342 2023-03-14 16:33:28 +09:00
parent 6fe6a12d04
commit 411e7f73b2
2 changed files with 5 additions and 5 deletions

View File

@ -833,11 +833,11 @@ def compute_loss(
batch_idx_train = params.batch_idx_train batch_idx_train = params.batch_idx_train
warm_step = params.warm_step warm_step = params.warm_step
#texts = batch["supervisions"]["text"] texts = batch["supervisions"]["text"]
texts = [] #texts = []
for cut in supervisions['cut']: #for cut in supervisions['cut']:
utt_id = cut.id # utt_id = cut.id
texts.append(pl_texts[utt_id]) # texts.append(pl_texts[utt_id])
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)