fix: quad training

2024-10-14 10:15:19 +08:00 · 2024-10-14 10:15:19 +08:00 · 4be220c4b4
parent a355524157
commit 4be220c4b4
1 changed files with 1 additions and 0 deletions
--- a/utils/dataloaders.py
+++ b/utils/dataloaders.py
@ -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,