diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-07-30 17:12:55 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | ea0147d9b90a0ee24c42c129a5ebdc7ef3c85876 (patch) | |
tree | 82e749032249e924309584178780f03d00d5c702 /examples/graph_squeezenet_v1_1.cpp | |
parent | eb84d6b82ba744295f5249d2171ee6537e09751a (diff) | |
download | ComputeLibrary-ea0147d9b90a0ee24c42c129a5ebdc7ef3c85876.tar.gz |
COMPMID-1188: Fix squeesenetv1.1 input shape
Change-Id: I39e354327a87ebe838af9f1cd57b5800517cf7ea
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/141964
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'examples/graph_squeezenet_v1_1.cpp')
-rw-r--r-- | examples/graph_squeezenet_v1_1.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/graph_squeezenet_v1_1.cpp b/examples/graph_squeezenet_v1_1.cpp index c0b5ff212d..631f123e63 100644 --- a/examples/graph_squeezenet_v1_1.cpp +++ b/examples/graph_squeezenet_v1_1.cpp @@ -72,7 +72,7 @@ public: std::unique_ptr<IPreprocessor> preprocessor = arm_compute::support::cpp14::make_unique<CaffePreproccessor>(mean_rgb); // Create input descriptor - const TensorShape tensor_shape = permute_shape(TensorShape(224U, 224U, 3U, 1U), DataLayout::NCHW, common_params.data_layout); + const TensorShape tensor_shape = permute_shape(TensorShape(227U, 227U, 3U, 1U), DataLayout::NCHW, common_params.data_layout); TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_layout(common_params.data_layout); // Set weights trained layout |