diff options
Diffstat (limited to 'python/pyarmnn/examples/example_utils.py')
| -rw-r--r-- | python/pyarmnn/examples/example_utils.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/pyarmnn/examples/example_utils.py b/python/pyarmnn/examples/example_utils.py index f4d1e4eb1f..5ef30f2331 100644 --- a/python/pyarmnn/examples/example_utils.py +++ b/python/pyarmnn/examples/example_utils.py @@ -73,7 +73,7 @@ def __create_network(model_file: str, backends: list, parser=None): def create_tflite_network(model_file: str, backends: list = ['CpuAcc', 'CpuRef']): - """Creates a network from an onnx model file. + """Creates a network from a tflite model file. Args: model_file (str): Path of the model file. @@ -92,7 +92,7 @@ def create_tflite_network(model_file: str, backends: list = ['CpuAcc', 'CpuRef'] def create_onnx_network(model_file: str, backends: list = ['CpuAcc', 'CpuRef']): - """Creates a network from a tflite model file. + """Creates a network from an onnx model file. Args: model_file (str): Path of the model file. |