from local

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

View File

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