Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/lpr.git" did not exist on "679ae84d6e88b324f5411e6ae030e15a019a8118"
Commit 7021ac1c authored by Taylor Robie's avatar Taylor Robie
Browse files

missed a not

parent f1efaf83
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/data_pipeline.py
View file @ 7021ac1c
...@@ -378,7 +378,7 @@ class BaseDataConstructor(threading.Thread): ...@@ -378,7 +378,7 @@ class BaseDataConstructor(threading.Thread):
self._current_epoch_order = np.empty(shape=(0,)) self._current_epoch_order = np.empty(shape=(0,))
self._shuffle_iterator = None self._shuffle_iterator = None
self._shuffle_with_forkpool = stream_files self._shuffle_with_forkpool = not stream_files
if stream_files: if stream_files:
self._shard_root = tempfile.mkdtemp(prefix="ncf_") self._shard_root = tempfile.mkdtemp(prefix="ncf_")
atexit.register(tf.gfile.DeleteRecursively, dirname=self._shard_root) atexit.register(tf.gfile.DeleteRecursively, dirname=self._shard_root)
......
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