from local

This commit is contained in:
dohe0342 2023-01-26 14:38:15 +09:00
parent f057f0376b
commit 6089ce6b2e
2 changed files with 1 additions and 0 deletions

View File

@ -1037,6 +1037,7 @@ 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)