diff --git a/README.md b/README.md index 622c341..5ea4e2f 100644 --- a/README.md +++ b/README.md @@ -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. @@ -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] diff --git a/sit4onnx/onnx_inference_test.py b/sit4onnx/onnx_inference_test.py index 0019ff9..e60643e 100644 --- a/sit4onnx/onnx_inference_test.py +++ b/sit4onnx/onnx_inference_test.py @@ -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]