Limit default --num-workers

tuxzz · Feb 7, 2023 · b28276d · b28276d
1 parent e09bd87
commit b28276d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nunif/training/trainer.py b/nunif/training/trainer.py
@@ -183,7 +183,7 @@ def create_trainer_default_parser():
     parser = argparse.ArgumentParser(
         add_help=False,
         formatter_class=argparse.ArgumentDefaultsHelpFormatter)
-    num_workers = cpu_count() - 2
+    num_workers = min(cpu_count() - 2, 8)
     if not num_workers > 0:
         num_workers = cpu_count()