Merge pull request #1737 from visi0nary/preprocessing_thread_fix
Preprocessing: use all available threads
This commit is contained in:
commit
b1f84ab7b4
@ -8,8 +8,8 @@ from multiprocessing.dummy import Pool
|
|||||||
from util.audio import audiofile_to_input_vector
|
from util.audio import audiofile_to_input_vector
|
||||||
from util.text import text_to_char_array
|
from util.text import text_to_char_array
|
||||||
|
|
||||||
def pmap(fun, iterable, threads=8):
|
def pmap(fun, iterable):
|
||||||
pool = Pool(threads)
|
pool = Pool()
|
||||||
results = pool.map(fun, iterable)
|
results = pool.map(fun, iterable)
|
||||||
pool.close()
|
pool.close()
|
||||||
return results
|
return results
|
||||||
|
Loading…
x
Reference in New Issue
Block a user