diff options
Diffstat (limited to 'examples/graph_resnet_v2_50.cpp')
-rw-r--r-- | examples/graph_resnet_v2_50.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/examples/graph_resnet_v2_50.cpp b/examples/graph_resnet_v2_50.cpp index b2c85e8938..c5c2e296f5 100644 --- a/examples/graph_resnet_v2_50.cpp +++ b/examples/graph_resnet_v2_50.cpp @@ -231,6 +231,11 @@ private: /** Main program for ResNetV2_50 * + * Model is based on: + * https://arxiv.org/abs/1603.05027 + * "Identity Mappings in Deep Residual Networks" + * Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun + * * @note To list all the possible arguments execute the binary appended with the --help option * * @param[in] argc Number of arguments |