Skip to content

S
seminar-breakout
Commit 9588d6ca authored by Patrick Wieschollek's avatar Patrick Wieschollek Committed by Yuxin Wu
Browse files
Options

fix 261 (#262)

parent 0ada5796
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/dataflow/common.py
View file @ 9588d6ca
...@@ -198,10 +198,10 @@ class FixedSizeData(ProxyDataFlow): ...@@ -198,10 +198,10 @@ class FixedSizeData(ProxyDataFlow):
cnt = 0 cnt = 0
while True: while True:
try: try:
dp = self.itr.next() dp = next(self.itr)
except StopIteration: except StopIteration:
self.itr = self.ds.get_data() self.itr = self.ds.get_data()
dp = self.itr.next() dp = next(self.itr)
cnt += 1 cnt += 1
yield dp yield dp
......
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