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
1e07957278
commit
a408082f14
Binary file not shown.
Binary file not shown.
@ -792,7 +792,6 @@ def compute_loss(
|
|||||||
batch: dict,
|
batch: dict,
|
||||||
is_training: bool,
|
is_training: bool,
|
||||||
decode: bool = False,
|
decode: bool = False,
|
||||||
pl_texts: dict = None,
|
|
||||||
) -> Tuple[Tensor, MetricsTracker]:
|
) -> Tuple[Tensor, MetricsTracker]:
|
||||||
"""
|
"""
|
||||||
Compute transducer loss given the model and its inputs.
|
Compute transducer loss given the model and its inputs.
|
||||||
@ -834,10 +833,6 @@ def compute_loss(
|
|||||||
warm_step = params.warm_step
|
warm_step = params.warm_step
|
||||||
|
|
||||||
texts = batch["supervisions"]["text"]
|
texts = batch["supervisions"]["text"]
|
||||||
#texts = []
|
|
||||||
#for cut in supervisions['cut']:
|
|
||||||
# utt_id = cut.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)
|
||||||
@ -1657,7 +1652,6 @@ def run_adapter(rank, world_size, args, wb=None):
|
|||||||
world_size=world_size,
|
world_size=world_size,
|
||||||
rank=rank,
|
rank=rank,
|
||||||
wb=wb,
|
wb=wb,
|
||||||
pl_texts=pl_texts,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if params.print_diagnostics:
|
if params.print_diagnostics:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user