From d167e61a23a54de529d51731fbe543ff4cec0d3c Mon Sep 17 00:00:00 2001
From: Luke Yeager <luke.yeager@gmail.com>
Date: Wed, 1 Jun 2016 09:50:57 -0700
Subject: 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
---
 include/caffe/net.hpp | 1 +
 1 file changed, 1 insertion(+)

(limited to 'include')

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() {}
 
-- 
cgit v1.2.3