diff options
Diffstat (limited to 'examples/graph_cl_shufflenet.cpp')
-rw-r--r-- | examples/graph_cl_shufflenet.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/examples/graph_cl_shufflenet.cpp b/examples/graph_cl_shufflenet.cpp index 70695bac56..0db34d6123 100644 --- a/examples/graph_cl_shufflenet.cpp +++ b/examples/graph_cl_shufflenet.cpp @@ -240,6 +240,11 @@ private: /** Main program for ShuffleNet * + * Model is based on: + * https://arxiv.org/abs/1707.01083 + * "ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices" + * Xiangyu Zhang, Xinyu Zhou, Mengxiao Lin, Jian Sun + * * @note To list all the possible arguments execute the binary appended with the --help option * * @param[in] argc Number of arguments |