made parallel optimize_restart responsive to ctrl+c

This commit is contained in:
Nicolo Fusi 2013-03-18 14:31:26 +00:00
parent b33d3e4c99
commit c44493077a

View file

@ -188,10 +188,10 @@ class model(parameterised):
"""
initial_parameters = self._get_params_transformed()
if parallel:
try:
jobs = []
pool = mp.Pool(processes=num_processes)
for i in range(Nrestarts):
@ -201,6 +201,10 @@ class model(parameterised):
pool.close() # signal that no more data coming in
pool.join() # wait for all the tasks to complete
except KeyboardInterrupt:
print "Ctrl+c received, terminating and joining pool."
pool.terminate()
pool.join()
for i in range(Nrestarts):
try: