Skip to content

Commit 5b8289f

Browse files
committed
reduced target_task_mem_mb in mulitpoc strategy
1 parent 094a81c commit 5b8289f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/imperandi/process/segment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -842,7 +842,7 @@ def main(args: argparse.Namespace) -> None:
842842
prefer_gpu=True,
843843
requested_workers=args.num_workers,
844844
start_method_hint=args.start_method,
845-
target_task_mem_mb=8000, # tune (TotalSegmentator can be heavy)
845+
target_task_mem_mb=6000, # tune (TotalSegmentator can be heavy)
846846
need_hard_timeouts=True,
847847
)
848848

0 commit comments

Comments
 (0)