From 0cc5c88075207f76ab071af3e9a5b45d6b9ed8a7 Mon Sep 17 00:00:00 2001 From: Kristian Hartikainen Date: Sat, 16 Feb 2019 01:12:59 -0800 Subject: [PATCH] [tune] Add number of trials to the trial runner logger (#4068) --- python/ray/tune/trial_runner.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/python/ray/tune/trial_runner.py b/python/ray/tune/trial_runner.py index 4f11d5226..93eaf1068 100644 --- a/python/ray/tune/trial_runner.py +++ b/python/ray/tune/trial_runner.py @@ -305,6 +305,16 @@ class TrialRunner(object): for local_dir in sorted({t.local_dir for t in self._trials}): messages.append("Result logdir: {}".format(local_dir)) + + num_trials_per_state = { + state: len(trials) + for state, trials in states.items() + } + total_number_of_trials = sum(num_trials_per_state.values()) + messages.append("Number of trials: {} ({})" + "".format(total_number_of_trials, + num_trials_per_state)) + for state, trials in sorted(states.items()): limit = limit_per_state[state] messages.append("{} trials:".format(state))