Skip to content

GitLab

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

Fix crash on single-core systems. (#4957)

parent e6353fe5
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/recommendation/data_preprocessing.py
View file @ abc62005
...@@ -327,7 +327,8 @@ def construct_cache(dataset, data_dir, num_data_readers): ...@@ -327,7 +327,8 @@ def construct_cache(dataset, data_dir, num_data_readers):
data during training. data during training.
""" """
cache_paths = rconst.Paths(data_dir=data_dir) cache_paths = rconst.Paths(data_dir=data_dir)
num_data_readers = num_data_readers or int(multiprocessing.cpu_count() / 2) num_data_readers = (num_data_readers or int(multiprocessing.cpu_count() / 2)
or 1)
approx_num_shards = int(movielens.NUM_RATINGS[dataset] approx_num_shards = int(movielens.NUM_RATINGS[dataset]
// rconst.APPROX_PTS_PER_TRAIN_SHARD) or 1 // rconst.APPROX_PTS_PER_TRAIN_SHARD) or 1
......
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