diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py index e3f426cc1..db92bc8d5 100644 --- a/funasr/train_utils/trainer_ds.py +++ b/funasr/train_utils/trainer_ds.py @@ -712,8 +712,8 @@ class Trainer: "data_split_i": kwargs.get("data_split_i", 0), "data_split_num": kwargs.get("data_split_num", 1), "log_step": batch_idx + kwargs.get("start_step", 0), - "batch_total": batch_idx, - "step_in_epoch": batch_idx, + "batch_total": batch_idx + 1, + "step_in_epoch": batch_idx + 1, "lr": 0.0, }