from local
This commit is contained in:
parent
8da1068fd8
commit
5d6991b536
Binary file not shown.
@ -832,8 +832,7 @@ 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 = batch["supervisions"]["greedy pseudo text"]
|
|
||||||
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)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user