diff --git a/utils/datasets.py b/utils/datasets.py index 2a9c961e7c3b..b01201f46dd3 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -60,7 +60,6 @@ def create_dataloader(path, imgsz, batch_size, stride, opt, hyp=None, augment=Fa batch_size=batch_size, num_workers=nw, sampler=train_sampler, - drop_last=(train_sampler is not None), pin_memory=True, collate_fn=LoadImagesAndLabels.collate_fn) return dataloader, dataset