From 68ffe662ada805b7adf8f4e462154764a22df22d Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 14 Sep 2023 14:12:37 -0700 Subject: [PATCH] Update default # workers from 1 -> 4 --- src/training/params.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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."