diff --git a/src/training/params.py b/src/training/params.py index c33312f80..18ed9364d 100644 --- 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."