Renamed calculation.

Author: finbarrtimbers

open_instruct/grpo_fast.py

@@ -2241,7 +2241,7 @@ def generate_thread(args, vllm_engines, resume_training_step, stop_event, genera
     logger.info("[Generate Thread] 🛑 Stopping generation thread")
 
 
-def calculate_mfu_mbu(
+def calculate_utilization_metrics(
     model_dims: utils.ModelDims, packed_data: PackedData, train_duration: float, args: Args
 ) -> dict[str, float]:
     """Calculate Model FLOPs Utilization (MFU) for training.
@@ -2374,7 +2374,7 @@ def one_training_step(
     total_time = time.perf_counter() - start_time
 
     # Calculate MFU and MBU
-    utilization_metrics = calculate_mfu_mbu(model_dims, packed_data, train_timer.duration, args)
+    utilization_metrics = calculate_utilization_metrics(model_dims, packed_data, train_timer.duration, args)
 
     metrics = {
         "episode": episode,