Bug fixes and minor features

2024-04-25 06:14:31 -06:00
parent 5a70b7f38d
commit 5da3613e0b
12 changed files with 218 additions and 31 deletions
--- a/toolkit/data_loader.py
+++ b/toolkit/data_loader.py
@@ -585,7 +585,7 @@ def get_dataloader_from_datasets(
            drop_last=False,
            shuffle=True,
            collate_fn=dto_collation,  # Use the custom collate function
-            num_workers=4
+            num_workers=8
        )
    else:
        data_loader = DataLoader(