Merge branch 'master' into 'master'
make DataIterator compatible for python 2 See merge request !22
This commit is contained in:
commit
612a0f32d7
|
@ -1,3 +1,4 @@
|
|||
import six
|
||||
from .sampler import SequentialSampler, RandomSampler, BatchSampler
|
||||
|
||||
|
||||
|
@ -84,7 +85,11 @@ class DataIterator(object):
|
|||
return minibatch
|
||||
|
||||
def _next_index(self):
|
||||
return next(self._sampler_iter)
|
||||
if six.PY3:
|
||||
return next(self._sampler_iter)
|
||||
else:
|
||||
# six.PY2
|
||||
return self._sampler_iter.next()
|
||||
|
||||
def __len__(self):
|
||||
return len(self._index_sampler)
|
||||
|
|
Loading…
Reference in New Issue