mirror of
https://github.com/JoergFranke/ADNC.git
synced 2024-11-17 13:58:03 +08:00
bugfix in BatchGenerator, iter should return self not next(self)
This commit is contained in:
parent
b2ae8de21e
commit
390fb1f62e
@ -62,7 +62,7 @@ class BatchGenerator():
|
||||
self.order = self.data_set.rng.permutation(self.order)
|
||||
|
||||
def __iter__(self):
|
||||
return next(self)
|
||||
return self
|
||||
|
||||
def __next__(self):
|
||||
"""
|
||||
|
Loading…
Reference in New Issue
Block a user