Skip to content

S
seminar-breakout
Commit e1c0102e authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix bug in locallyshuffledata initialization

parent 8403a009
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
tensorpack/dataflow/common.py
View file @ e1c0102e
...@@ -332,8 +332,8 @@ class LocallyShuffleData(ProxyDataFlow, RNGDataFlow): ...@@ -332,8 +332,8 @@ class LocallyShuffleData(ProxyDataFlow, RNGDataFlow):
dp = next(self.ds_itr) dp = next(self.ds_itr)
for _ in range(self.nr_reuse): for _ in range(self.nr_reuse):
self.q.append(dp) self.q.append(dp)
for _ in range(self.q.maxlen - len(self.q)):
try: try:
while self.q.maxlen > len(self.q):
add_next() add_next()
except StopIteration: except StopIteration:
logger.error("LocallyShuffleData: cache_size is larger than the size of ds!") logger.error("LocallyShuffleData: cache_size is larger than the size of ds!")
......
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