Skip to content

Commit 045de53

Browse files
committed
Benchmark pytorch: add single_animal arg
1 parent e03fb1d commit 045de53

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

dlclive/benchmark_pytorch.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def benchmark(
8989
model_path: str,
9090
model_type: str,
9191
device: str,
92+
single_animal: bool,
9293
precision: str = "FP32",
9394
display=True,
9495
pcutoff=0.5,
@@ -156,6 +157,7 @@ def benchmark(
156157
dlc_live = DLCLive(
157158
model_path=model_path,
158159
model_type=model_type,
160+
single_animal=single_animal,
159161
device=device,
160162
display=display,
161163
resize=resize,

0 commit comments

Comments
 (0)