diff options
author | SiCongLi <sicong.li@arm.com> | 2021-06-04 10:47:07 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-06-08 17:36:27 +0000 |
commit | 4d7fff1e8668b2efdf7412ce2c8c0abf5fccd3b6 (patch) | |
tree | 870d7d6c307489175d628534bafa92e225e8271b /examples/graph_lenet.cpp | |
parent | 5e53fc65a6d7a966541c3020593eefc5222ef914 (diff) | |
download | ComputeLibrary-4d7fff1e8668b2efdf7412ce2c8c0abf5fccd3b6.tar.gz |
Fix missing network references
* Add missing references to LeNet
* Remove mnist example, which is an internally developed network
Resolves: COMPMID-4559
Signed-off-by: SiCongLi <sicong.li@arm.com>
Change-Id: I139354890f1a9acb54a87add6895262102b3b8de
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5782
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'examples/graph_lenet.cpp')
-rw-r--r-- | examples/graph_lenet.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/examples/graph_lenet.cpp b/examples/graph_lenet.cpp index 6560a980cc..1bcd95fb58 100644 --- a/examples/graph_lenet.cpp +++ b/examples/graph_lenet.cpp @@ -132,6 +132,14 @@ private: /** Main program for LeNet * + * Model is based on: + * http://yann.lecun.com/exdb/publis/pdf/lecun-98.pdf + * "Gradient-Based Learning Applied to Document Recognition" + * Yann LeCun, Léon Bottou, Yoshua Bengio, and Patrick Haffner + * + * The original model uses tanh instead of relu activations. However the use of relu activations in lenet has been + * widely adopted to improve accuracy.* + * * @note To list all the possible arguments execute the binary appended with the --help option * * @param[in] argc Number of arguments |