Skip to content

Commit 2e27587

Browse files
authored
Super tiny rename total_train_time to step_time for clarity (#629)
1 parent 7ddb4a9 commit 2e27587

File tree

1 file changed

+1
-1
lines changed
  • slime/backends/megatron_utils

1 file changed

+1
-1
lines changed

slime/backends/megatron_utils/data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -452,7 +452,7 @@ def log_perf_data(rollout_id: int, args: Namespace) -> None:
452452
if "perf/train_wait_time" in log_dict and "perf/train_time" in log_dict:
453453
total_time = log_dict["perf/train_wait_time"] + log_dict["perf/train_time"]
454454
if total_time > 0:
455-
log_dict["perf/total_train_time"] = total_time
455+
log_dict["perf/step_time"] = total_time
456456
log_dict["perf/wait_time_ratio"] = log_dict["perf/train_wait_time"] / total_time
457457

458458
print(f"perf {rollout_id}: {log_dict}")

0 commit comments

Comments
 (0)