from local

This commit is contained in:
dohe0342 2022-12-10 13:56:19 +09:00
parent 088a522667
commit ca612e1c01
2 changed files with 4 additions and 0 deletions

View File

@ -926,6 +926,10 @@ def train_one_epoch(
cur_batch_idx = params.get("cur_batch_idx", 0)
if params.multi_optim:
optimizer_enc, optimizer_dec = optimizer[0], optimizer[1]
scheduler_enc, scheduler_dec = scheduler[0], scheduler[1]
for batch_idx, batch in enumerate(train_dl):
if batch_idx < cur_batch_idx:
continue