nit

mlcommons · Dec 7, 2023 · 33ecf8e · 33ecf8e
1 parent bcf1ca0
commit 33ecf8e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -377,12 +377,15 @@ def train_once(
           train_state['test_goal_reached'] = (
               workload.has_reached_test_target(latest_eval_result) or
               train_state['test_goal_reached'])
+
           # Save last eval time.
           eval_end_time = get_time()
           train_state['last_eval_time'] = eval_end_time
+
           # Accumulate eval time.
           train_state[
               'accumulated_eval_time'] += eval_end_time - eval_start_time
+
           # Add times to eval results for logging.
           latest_eval_result['score'] = (
               train_state['accumulated_submission_time'])