Skip to content

Commit

Permalink
README updated
Browse files Browse the repository at this point in the history
  • Loading branch information
PINTO0309 committed May 14, 2022
1 parent 63dc062 commit e5dd83c
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,8 @@ optional arguments:
--batch_size BATCH_SIZE
Value to be substituted if input batch size is undefined.
This is ignored if the input dimensions are all of static size.
Also ignored if input_numpy_file_paths_for_testing or
numpy_ndarrays_for_testing is specified.

--test_loop_count TEST_LOOP_COUNT
Number of times to run the test.
Expand Down Expand Up @@ -103,6 +105,8 @@ inference(
batch_size: Optional[int]
Value to be substituted if input batch size is undefined.
This is ignored if the input dimensions are all of static size.
Also ignored if input_numpy_file_paths_for_testing or
numpy_ndarrays_for_testing is specified.
Default: 1

test_loop_count: Optional[int]
Expand Down
2 changes: 2 additions & 0 deletions sit4onnx/onnx_inference_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,6 +96,8 @@ def inference(
batch_size: Optional[int]
Value to be substituted if input batch size is undefined.\n\
This is ignored if the input dimensions are all of static size.\n\
Also ignored if input_numpy_file_paths_for_testing or\n\
numpy_ndarrays_for_testing is specified.\n\
Default: 1
test_loop_count: Optional[int]
Expand Down

0 comments on commit e5dd83c

Please sign in to comment.