Skip to content

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

fix LMDBSerializer

parent bf8acbfb
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tensorpack/dataflow/format.py
View file @ ffd9eaeb
......@@ -145,8 +145,7 @@ class LMDBData(RNGDataFlow):
with self._guard:
if not self._shuffle:
c = self._txn.cursor()
while next(c):
k, v = c.item()
for k, v in c:
if k != b'__keys__':
yield [k, v]
else:
......
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