fix(exp): change num_workers value

penn513 · Jul 19, 2021 · 8506f7e · 8506f7e
1 parent 929754a
commit 8506f7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/yolox/exp/yolox_base.py b/yolox/exp/yolox_base.py
@@ -23,7 +23,8 @@ def __init__(self):
         self.width = 1.00
 
         # ---------------- dataloader config ---------------- #
-        self.data_num_workers = 6
+        # set worker to 4 for shorter dataloader init time
+        self.data_num_workers = 4
         self.input_size = (640, 640)
         self.random_size = (14, 26)
         self.train_ann = "instances_train2017.json"