from local

This commit is contained in:
dohe0342 2023-04-11 14:30:05 +09:00
parent 85697a7370
commit fef51784e4
2 changed files with 2 additions and 0 deletions

View File

@ -1055,6 +1055,8 @@ def train_one_epoch(
is_training=True,
decode = True if batch_idx % params.decode_interval == 0 else False,
)
print(loss_info)
try: loss_info.reduce(loss.device)
except: pass