mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
be5f456790
commit
5a3e26e1e5
Binary file not shown.
@ -852,6 +852,7 @@ def compute_loss(
|
|||||||
|
|
||||||
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)
|
||||||
|
print(y)
|
||||||
|
|
||||||
with torch.set_grad_enabled(is_training):
|
with torch.set_grad_enabled(is_training):
|
||||||
simple_loss, pruned_loss, ctc_output = model(
|
simple_loss, pruned_loss, ctc_output = model(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user