From 33ecf8e375900e2f22d9e7a92e835c2d2b5f2f99 Mon Sep 17 00:00:00 2001 From: Boyuan Feng Date: Thu, 7 Dec 2023 14:46:00 -0800 Subject: [PATCH] nit --- submission_runner.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/submission_runner.py b/submission_runner.py index df7be7a86..31453c736 100644 --- 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'])