Update default # workers from 1 -> 4

mlfoundations · Sep 14, 2023 · 68ffe66 · 68ffe66
1 parent f692ec9
commit 68ffe66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/training/params.py b/src/training/params.py
@@ -120,7 +120,7 @@ def parse_args(args):
         help="Optional identifier for the experiment when storing logs. Otherwise use current time.",
     )
     parser.add_argument(
-        "--workers", type=int, default=1, help="Number of dataloader workers per GPU."
+        "--workers", type=int, default=4, help="Number of dataloader workers per GPU."
     )
     parser.add_argument(
         "--batch-size", type=int, default=64, help="Batch size per GPU."