Merge pull request #184 from zuoxingdong/step_info_trajectory

update run_experiment
zuoxingdong · May 11, 2019 · 93d5d6b · 93d5d6b
2 parents e5994df + e77635a
commit 93d5d6b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lagom/experiment/run_experiment.py b/lagom/experiment/run_experiment.py
@@ -11,6 +11,7 @@
 from lagom.utils import yaml_dump
 from lagom.utils import ask_yes_or_no
 from lagom.utils import timeit
+from lagom.utils import color_str
 from lagom.utils import CloudpickleWrapper
 
 
@@ -132,4 +133,5 @@ def _run(job):
     else:
         with ProcessPoolExecutor(max_workers=max_workers) as executor:
             results = list(executor.map(CloudpickleWrapper(_run), jobs, chunksize=chunksize))
+    print(color_str(f'\nExperiment finished. Loggings are stored in {log_path.absolute()}. ', 'cyan', 'bold'))
     return results