We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7ddb4a9 commit 2e27587Copy full SHA for 2e27587
slime/backends/megatron_utils/data.py
@@ -452,7 +452,7 @@ def log_perf_data(rollout_id: int, args: Namespace) -> None:
452
if "perf/train_wait_time" in log_dict and "perf/train_time" in log_dict:
453
total_time = log_dict["perf/train_wait_time"] + log_dict["perf/train_time"]
454
if total_time > 0:
455
- log_dict["perf/total_train_time"] = total_time
+ log_dict["perf/step_time"] = total_time
456
log_dict["perf/wait_time_ratio"] = log_dict["perf/train_wait_time"] / total_time
457
458
print(f"perf {rollout_id}: {log_dict}")
0 commit comments