From 0b8568ebfb484b9155d19c567e890ff52776544d Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Wed, 15 Feb 2023 15:11:37 +0100
Subject: Update

---
 data/csv.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'data')

diff --git a/data/csv.py b/data/csv.py
index 233f5d8..619452e 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -403,6 +403,7 @@ class VlpnDataset(IterableDataset):
             if len(batch) >= batch_size:
                 yield batch
                 batch = []
+                continue
 
             bucket_mask = mask.logical_and(self.bucket_assignments == bucket)
             bucket_items = self.bucket_items[bucket_mask]
-- 
cgit v1.2.3-70-g09d2