Skip to content

Commit ffd8083

Browse files
committed
fix docstring
1 parent edf7d0d commit ffd8083

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/deepsparse/image_classification/validation_script.py

+4
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,10 @@
3030
--image-size, --image_size INTEGER
3131
integer size to evaluate images at (will be
3232
reshaped to square shape) [default: 224]
33+
--batch-size, --batch_size INTEGER
34+
Test batch size, must divide the dataset
35+
evenly, else last batch will be dropped
36+
[default: 1]
3337
--num-cores, --num_cores INTEGER
3438
Number of CPU cores to run deepsparse with,
3539
default is all available

0 commit comments

Comments
 (0)