from local

This commit is contained in:
dohe0342 2022-12-10 13:59:03 +09:00
parent 766ed65609
commit c7bfa2c95a
2 changed files with 2 additions and 2 deletions

View File

@ -934,8 +934,8 @@ def train_one_epoch(
if batch_idx < cur_batch_idx: if batch_idx < cur_batch_idx:
continue continue
cur_batch_idx = batch_idx cur_batch_idx = batch_idx
params.batch_idx_train += 1 if batch_idx % params.accum_grads == 0: params.batch_idx_train += 1
batch_size = len(batch["supervisions"]["text"]) batch_size = len(batch["supervisions"]["text"])
try: try: