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

abc62005 · Reed · Taylor Robie · e6353fe5 · abc62005
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

official/recommendation/data_preprocessing.py official/recommendation/data_preprocessing.py +2 -1

未找到文件。
--- a/official/recommendation/data_preprocessing.py
+++ b/official/recommendation/data_preprocessing.py
@@ -327,7 +327,8 @@ def construct_cache(dataset, data_dir, num_data_readers):
      data during training.
  """
  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]
                          // rconst.APPROX_PTS_PER_TRAIN_SHARD) or 1