Skip to content

Dynamic setting of input batch size and support for multiple nets in Matlab wrapper #2048

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
kencoken wants to merge 5 commits into from
Closed

Dynamic setting of input batch size and support for multiple nets in Matlab wrapper #2048

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
e97daab
Added override parameter for network input count
kencoken Oct 24, 2014
981f87e
Updated matcaffe to support net handle array and retrieval of blob sizes
kencoken Feb 6, 2015
a266232
Bugfix to get_blob_size endpoint of mex file
kencoken Feb 10, 2015
f93b48e
Bugfixes
kencoken Mar 5, 2015
ba0afbd
Fix lint errors
kencoken Mar 6, 2015
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 8 additions & 3 deletions include/caffe/net.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,12 +23,15 @@ namespace caffe {
template <typename Dtype>
class Net {
public:
explicit Net(const NetParameter& param);
explicit Net(const string& param_file, Phase phase);
explicit Net(const NetParameter& param,
const size_t force_input_count = 0);
explicit Net(const string& param_file, Phase phase,
const size_t force_input_count = 0);
virtual ~Net() {}

/// @brief Initialize a network with a NetParameter.
void Init(const NetParameter& param);
void Init(const NetParameter& param,
const size_t force_input_count = 0);

/**
* @brief Run Forward with the input Blob%s already fed separately.
Expand Down Expand Up @@ -256,6 +259,8 @@ class Net {
/// Whether to compute and display debug info for the net.
bool debug_info_;

size_t force_input_count_;

DISABLE_COPY_AND_ASSIGN(Net);
};

Expand Down
Loading