from local

This commit is contained in:
dohe0342 2023-04-18 13:59:38 +09:00
parent 1ea3e66e24
commit b834fe6031
3 changed files with 2 additions and 0 deletions

View File

@ -1002,6 +1002,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