from local

This commit is contained in:
dohe0342 2023-03-29 22:07:05 +09:00
parent b2530f8489
commit 3b140f3576
2 changed files with 2 additions and 0 deletions

View File

@ -1018,6 +1018,8 @@ def train_one_epoch(
scheduler_enc, scheduler_dec = scheduler[0], scheduler[1]
for batch_idx, batch in enumerate(train_dl):
if params.batch_idx_train > params.num_updates:
break
if batch_idx < cur_batch_idx:
continue
cur_batch_idx = batch_idx