From 47c14aff79bb902482b0a953a55d31bf130c7b04 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 20 五月 2024 16:20:45 +0800
Subject: [PATCH] add
---
funasr/train_utils/trainer_ds.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/train_utils/trainer_ds.py b/funasr/train_utils/trainer_ds.py
index e3f426c..db92bc8 100644
--- a/funasr/train_utils/trainer_ds.py
+++ b/funasr/train_utils/trainer_ds.py
@@ -712,8 +712,8 @@
"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,
}
--
Gitblit v1.9.1