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:
continue
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"])
try: