updates
This commit is contained in:
parent
72afa02272
commit
8ebb4da5cc
|
@ -104,7 +104,6 @@ class LoadImagesAndLabels: # for training
|
||||||
self.batch_size = batch_size
|
self.batch_size = batch_size
|
||||||
self.img_size = img_size
|
self.img_size = img_size
|
||||||
self.augment = augment
|
self.augment = augment
|
||||||
iter(self)
|
|
||||||
|
|
||||||
def __iter__(self):
|
def __iter__(self):
|
||||||
self.count = -1
|
self.count = -1
|
||||||
|
|
Loading…
Reference in New Issue