Merge pull request #248 from Mark-ZhouWX/yolox

fix OOM of run_eval mode
mindspore-lab · Nov 24, 2023 · 0983315 · 0983315
2 parents 54c3bcb + 66385f9
commit 0983315
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -183,7 +183,7 @@ def train(args):
             rank_size=args.rank_size,
             shuffle=False,
             drop_remainder=False,
-            num_parallel_workers=args.data.num_parallel_workers,
+            num_parallel_workers=1,
             python_multiprocessing=True,
         )
     else: