from local

This commit is contained in:
dohe0342 2022-12-10 15:23:48 +09:00
parent 0e0b4cc37f
commit 98be3eb134
2 changed files with 1 additions and 1 deletions

View File

@ -967,7 +967,7 @@ def train_one_epoch(
sp=sp,
batch=batch,
is_training=True,
decode = True if batch_idx % params.decode_interval == 0 else False
decode = True if batch_idx % params.decode_interval == 0 else False,
)
loss_info.reduce(loss.device)