We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e03fb1d commit 045de53Copy full SHA for 045de53
dlclive/benchmark_pytorch.py
@@ -89,6 +89,7 @@ def benchmark(
89
model_path: str,
90
model_type: str,
91
device: str,
92
+ single_animal: bool,
93
precision: str = "FP32",
94
display=True,
95
pcutoff=0.5,
@@ -156,6 +157,7 @@ def benchmark(
156
157
dlc_live = DLCLive(
158
model_path=model_path,
159
model_type=model_type,
160
+ single_animal=single_animal,
161
device=device,
162
display=display,
163
resize=resize,
0 commit comments