Skip to content

Commit

Permalink
fix(demo): Add default number of CPU threads in the layout demo.
Browse files Browse the repository at this point in the history
Signed-off-by: Nikos Livathinos <[email protected]>
  • Loading branch information
nikos-livathinos committed Dec 8, 2024
1 parent 76fbc48 commit cda8a4d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion demo/demo_layout_predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ def main(args):
"-d", "--device", required=False, default="cpu", help="One of [cpu, cuda, mps]"
)
parser.add_argument(
"-n", "--num_threads", required=False, default=None, help="Number of threads"
"-n", "--num_threads", required=False, default=4, help="Number of threads"
)
parser.add_argument(
"-i",
Expand Down

0 comments on commit cda8a4d

Please sign in to comment.