From 47c14aff79bb902482b0a953a55d31bf130c7b04 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Mon, 20 May 2024 16:20:45 +0800 Subject: [PATCH] add --- funasr/train_utils/trainer_ds.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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, }