from local

This commit is contained in:
dohe0342 2023-03-14 17:26:15 +09:00
parent 294fc81346
commit 2ee95d4d00
2 changed files with 0 additions and 2 deletions

View File

@ -975,7 +975,6 @@ def train_one_epoch(
world_size: int = 1,
rank: int = 0,
wb = None,
pl_texts = None,
) -> None:
"""Train the model for one epoch.
@ -1035,7 +1034,6 @@ def train_one_epoch(
batch=batch,
is_training=True,
decode = True if batch_idx % params.decode_interval == 0 else False,
pl_texts = pl_texts,
)
loss_info.reduce(loss.device)