prevent warning for max workers

jiezhangGt · May 6, 2022 · 7e93625 · 7e93625
1 parent 636a4aa
commit 7e93625
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/comet/cli/compare.py b/comet/cli/compare.py
@@ -52,7 +52,6 @@
   --print_cache_info    Print information about COMET cache. (default: False)
 """
 import json
-import multiprocessing
 import os
 from itertools import combinations
 from typing import Dict, Generator, List, Tuple, Union
@@ -341,7 +340,7 @@ def get_cfg() -> Namespace:
         "--num_workers",
         help="Number of workers to use when loading data.",
         type=int,
-        default=multiprocessing.cpu_count(),
+        default=2,
     )
     parser.add_argument(
         "--disable_bar", action="store_true", help="Disables progress bar."