diff options
author | ZJaume <jzaragoza@prompsit.com> | 2022-07-05 16:00:05 +0300 |
---|---|---|
committer | ZJaume <jzaragoza@prompsit.com> | 2022-07-05 16:00:05 +0300 |
commit | f25b5188fc07bca09096388e1a8ae6fe97675a77 (patch) | |
tree | e52af919806291db086b36acaf89ca8a89fd6e90 | |
parent | 265c87cfaef7b9ede47ac7223fc90ac06cfc4297 (diff) |
Set inter/intra_op parallelism to 0 by default
There's no need of setting it to max cpu, TF will set an optimal value
-rw-r--r-- | bicleaner_ai/classify.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/bicleaner_ai/classify.py b/bicleaner_ai/classify.py index 18df8a8..7131448 100644 --- a/bicleaner_ai/classify.py +++ b/bicleaner_ai/classify.py @@ -42,7 +42,7 @@ def argument_parser(): groupO.add_argument("--scol", default=3 if not header else "src_text", type=check_positive if not header else str, help ="Source sentence column (starting in 1). The name of the field is expected instead of the position if --header is set") groupO.add_argument("--tcol", default=4 if not header else "trg_text", type=check_positive if not header else str, help ="Target sentence column (starting in 1). The name of the field is expected instead of the position if --header is set") groupO.add_argument('-b', '--block_size', type=int, default=1000, help="Sentence pairs per block") - groupO.add_argument('-p', '--processes', type=int, default=max(1, cpu_count()-1), help="Number of processes to use") + groupO.add_argument('-p', '--processes', type=int, default=0, help="Number of processes to use") groupO.add_argument('--batch_size', type=int, default=32, help="Sentence pairs per block") groupO.add_argument('--tmp_dir', default=gettempdir(), help="Temporary directory where creating the temporary files of this program") |