arm_compute v18.11
diff --git a/examples/graph_alexnet.cpp b/examples/graph_alexnet.cpp
index 124f672..cfa8e94 100644
--- a/examples/graph_alexnet.cpp
+++ b/examples/graph_alexnet.cpp
@@ -31,11 +31,7 @@
using namespace arm_compute::graph::frontend;
using namespace arm_compute::graph_utils;
-/** Example demonstrating how to implement AlexNet's network using the Compute Library's graph API
- *
- * @param[in] argc Number of arguments
- * @param[in] argv Arguments
- */
+/** Example demonstrating how to implement AlexNet's network using the Compute Library's graph API */
class GraphAlexnetExample : public Example
{
public:
@@ -58,12 +54,6 @@
return false;
}
- // Set default layout if needed
- if(!common_opts.data_layout->is_set() && common_params.target == Target::NEON)
- {
- common_params.data_layout = DataLayout::NCHW;
- }
-
// Checks
ARM_COMPUTE_EXIT_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "QASYMM8 not supported for this graph");
@@ -181,6 +171,11 @@
/** Main program for AlexNet
*
+ * Model is based on:
+ * https://papers.nips.cc/paper/4824-imagenet-classification-with-deep-convolutional-neural-networks
+ * "ImageNet Classification with Deep Convolutional Neural Networks"
+ * Alex Krizhevsky and Sutskever, Ilya and Hinton, Geoffrey E
+ *
* @note To list all the possible arguments execute the binary appended with the --help option
*
* @param[in] argc Number of arguments