from local

This commit is contained in:
dohe0342 2022-12-12 16:13:08 +09:00
parent 738c392a9d
commit bd6bb36680
2 changed files with 1 additions and 1 deletions

View File

@ -1082,7 +1082,7 @@ def train_one_epoch(
f"grad_scale is too small, exiting: {cur_grad_scale}"
)
if batch_idx % params.log_interval == 0:
if batch_idx % (params.log_interval*params.accum_grads) == 0:
if params.multi_optim:
cur_enc_lr = scheduler_enc.get_last_lr()[0]
cur_dec_lr = scheduler_dec.get_last_lr()[0]