Fix bug when enable `--quad` training option (#13355)
* fix: quad training * fix: quad training in segmentationpull/13365/head
parent
abdfbd68d1
commit
94a6245617
|
@ -210,6 +210,7 @@ def create_dataloader(
|
|||
shuffle=shuffle and sampler is None,
|
||||
num_workers=nw,
|
||||
sampler=sampler,
|
||||
drop_last=quad,
|
||||
pin_memory=PIN_MEMORY,
|
||||
collate_fn=LoadImagesAndLabels.collate_fn4 if quad else LoadImagesAndLabels.collate_fn,
|
||||
worker_init_fn=seed_worker,
|
||||
|
|
|
@ -75,6 +75,7 @@ def create_dataloader(
|
|||
shuffle=shuffle and sampler is None,
|
||||
num_workers=nw,
|
||||
sampler=sampler,
|
||||
drop_last=quad,
|
||||
pin_memory=True,
|
||||
collate_fn=LoadImagesAndLabelsAndMasks.collate_fn4 if quad else LoadImagesAndLabelsAndMasks.collate_fn,
|
||||
worker_init_fn=seed_worker,
|
||||
|
|
Loading…
Reference in New Issue