from local

This commit is contained in:
dohe0342 2022-12-10 14:59:56 +09:00
parent a56fb98700
commit c06244081c
2 changed files with 2 additions and 2 deletions

View File

@ -1038,8 +1038,8 @@ def train_one_epoch(
if batch_idx % params.log_interval == 0:
if params.multi_optim:
cur_enc_lr = scheduler.get_last_lr()[0]
cur_dec_lr = scheduler.get_last_lr()[0]
cur_enc_lr = scheduler_enc.get_last_lr()[0]
cur_dec_lr = scheduler_dec.get_last_lr()[0]
cur_grad_scale = scaler._scale.item() if params.use_fp16 else 1.0
logging.info(