aboutsummaryrefslogtreecommitdiff
path: root/examples/graph_alexnet.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'examples/graph_alexnet.cpp')
-rw-r--r--examples/graph_alexnet.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/examples/graph_alexnet.cpp b/examples/graph_alexnet.cpp
index c607672e5b..cfa8e94d10 100644
--- a/examples/graph_alexnet.cpp
+++ b/examples/graph_alexnet.cpp
@@ -171,6 +171,11 @@ private:
/** 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