Merge branch 'main' of github.com:Wadaboa/titanet into main

Wadaboa · Wadaboa · commit 7b77053051d8 · 2022-10-30T23:04:56.000Z
diff --git a/src/datasets.py b/src/datasets.py
@@ -41,6 +41,7 @@ def get_dataloader(
         drop_last=False,
         generator=generator,
         persistent_workers=True,
+        drop_last=True,
     )
 
 
