Skip to content

Commit

Permalink
Lower default num_workers
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanklut committed Jun 25, 2024
1 parent 4ac0d03 commit e3e4816
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions run.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ def get_arguments() -> argparse.Namespace:
io_args.add_argument("-o", "--output", help="Output folder", type=str, required=True)

parser.add_argument("-w", "--whitelist", nargs="+", help="Input folder", type=str, action="extend")
parser.add_argument("--num_workers", help="Number of workers to use", type=int, default=16)
parser.add_argument("--num_workers", help="Number of workers to use", type=int, default=4)

args = parser.parse_args()

Expand Down Expand Up @@ -244,7 +244,7 @@ def __init__(
input_paths: str | Path | Sequence[str | Path],
output_dir: str | Path,
output_page: OutputPageXML,
num_workers: int = 16,
num_workers: int = 4,
):
"""
Extension on the predictor that actually saves the part on the prediction we current care about: the semantic segmentation as pageXML
Expand Down

0 comments on commit e3e4816

Please sign in to comment.