diff options
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 |