diff options
| author | Luke Yeager <luke.yeager@gmail.com> | 2016-06-01 09:50:57 -0700 |
|---|---|---|
| committer | Luke Yeager <luke.yeager@gmail.com> | 2016-06-03 15:28:57 -0700 |
| commit | d167e61a23a54de529d51731fbe543ff4cec0d3c (patch) | |
| tree | 97842e2979f349d50f969e989a56e84b653a6249 /include | |
| parent | df412ac0da3e2e7eb194f0c16842fd126496d90d (diff) | |
| download | caffeonacl-d167e61a23a54de529d51731fbe543ff4cec0d3c.tar.gz caffeonacl-d167e61a23a54de529d51731fbe543ff4cec0d3c.tar.bz2 caffeonacl-d167e61a23a54de529d51731fbe543ff4cec0d3c.zip | |
Add level and stages to Net constructor
This internal functionality will be exposed through the various
interfaces in subsequent commits
Also adds C++ tests for all-in-one nets
Diffstat (limited to 'include')
| -rw-r--r-- | include/caffe/net.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/include/caffe/net.hpp b/include/caffe/net.hpp index 0addb3c2..493bdf29 100644 --- a/include/caffe/net.hpp +++ b/include/caffe/net.hpp @@ -25,6 +25,7 @@ class Net { public: explicit Net(const NetParameter& param, const Net* root_net = NULL); explicit Net(const string& param_file, Phase phase, + const int level = 0, const vector<string>* stages = NULL, const Net* root_net = NULL); virtual ~Net() {} |