From 721baf4298483dc1df170783150f3dbd70ef9b07 Mon Sep 17 00:00:00 2001 From: Fang Zhang Date: Fri, 28 Apr 2023 16:25:08 +0800 Subject: [PATCH] fix step_time --- applications/DeepSpeed-Chat/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/applications/DeepSpeed-Chat/train.py b/applications/DeepSpeed-Chat/train.py index b57d80759..dae5273cd 100644 --- a/applications/DeepSpeed-Chat/train.py +++ b/applications/DeepSpeed-Chat/train.py @@ -194,7 +194,7 @@ def main(args): cmd = get_cmd(args, step_num) launch_cmd(args, step_num, cmd) - step_time = int(time.time() - start_time) + step_time = int(time.time() - step_start_time) time_str = str(datetime.timedelta(seconds=step_time)) print(f"---=== Finished Step {step_num} in {time_str} ===---")