Skip to content

GitLab

Commit c242705b authored by Reed's avatar Reed Committed by Taylor Robie
Browse files

Fix another crash on single-core systems. (#4962)

parent c708c1b4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/data_preprocessing.py
View file @ c242705b
...@@ -395,7 +395,7 @@ def instantiate_pipeline(dataset, data_dir, batch_size, eval_batch_size, ...@@ -395,7 +395,7 @@ def instantiate_pipeline(dataset, data_dir, batch_size, eval_batch_size,
# By limiting the number of workers we guarantee that the worker # By limiting the number of workers we guarantee that the worker
# pool underlying the training generation doesn't starve other processes. # pool underlying the training generation doesn't starve other processes.
num_workers = int(multiprocessing.cpu_count() * 0.75) num_workers = int(multiprocessing.cpu_count() * 0.75) or 1
subproc_args = popen_helper.INVOCATION + [ subproc_args = popen_helper.INVOCATION + [
"--data_dir", data_dir, "--data_dir", data_dir,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment