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
2e61476b01
commit
41551bacf1
Binary file not shown.
Binary file not shown.
@ -995,7 +995,6 @@ def train_one_epoch(
|
||||
world_size: int = 1,
|
||||
rank: int = 0,
|
||||
wb = None,
|
||||
prompt = None,
|
||||
) -> None:
|
||||
"""Train the model for one epoch.
|
||||
|
||||
@ -1057,7 +1056,6 @@ def train_one_epoch(
|
||||
batch=batch,
|
||||
is_training=True,
|
||||
decode = True if batch_idx % params.decode_interval == 0 else False,
|
||||
prompt=prompt,
|
||||
)
|
||||
|
||||
try: loss_info.reduce(loss.device)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user