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'])