Remove opt from create_dataloader()` (#3552)
This commit is contained in:
1 parent
0cfc5b2c18
commit
958ab92dc1
3 files changed
+13
-12
No files matched your search
@@ -88,7 +88,7 @@ def test(data,
|
||||
if device.type != 'cpu':
|
||||
model(torch.zeros(1, 3, imgsz, imgsz).to(device).type_as(next(model.parameters()))) # run once
|
||||
task = opt.task if opt.task in ('train', 'val', 'test') else 'val' # path to train/val/test images
|
||||
dataloader = create_dataloader(data[task], imgsz, batch_size, gs, opt, pad=0.5, rect=True,
|
||||
dataloader = create_dataloader(data[task], imgsz, batch_size, gs, single_cls, pad=0.5, rect=True,
|
||||
prefix=colorstr(f'{task}: '))[0]
|
||||
|
||||
seen = 0
|
||||
|
||||
Reference in new issue
Block a user