diff options
Diffstat (limited to 'tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp')
| -rw-r--r-- | tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp b/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp index 3d57666b58..44bbb70989 100644 --- a/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp +++ b/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp @@ -50,9 +50,9 @@ int main(int argc, char* argv[]) // Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions. retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>( argc, argv, - "mobilenet_v1_1.0_224_fp32.pb", // model name + "mobilenet_v1_1.0_224_frozen.pb", // model name true, // model is binary - "input", "output", // input and output tensor names + "input", "MobilenetV1/Predictions/Reshape_1", // input and output tensor names { 0, 1, 2 }, // test images to test with as above [&imageSet](const char* dataDir, const ModelType&) { // This creates a 224x224x3 NHWC float tensor to pass to Armnn |