from local

This commit is contained in:
dohe0342 2023-04-11 15:37:10 +09:00
parent 8527f5d8c9
commit 76cc492dd9
2 changed files with 1 additions and 0 deletions

View File

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