This commit is contained in:
Xianjie Chen 2015-01-22 19:27:24 -08:00
Родитель 64b68b274c
Коммит f4b386537e
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -52,7 +52,7 @@ static int init_key = -2;
// input and outputs a cell array.
static mxArray* do_forward(const mxArray* const bottom) {
vector<Blob<float>*>& input_blobs = net_->input_blobs();
const vector<Blob<float>*>& input_blobs = net_->input_blobs();
if (static_cast<unsigned int>(mxGetDimensions(bottom)[0]) !=
input_blobs.size()) {
mex_error("Invalid input size");
@ -112,8 +112,8 @@ static mxArray* do_forward(const mxArray* const bottom) {
}
static mxArray* do_backward(const mxArray* const top_diff) {
vector<Blob<float>*>& output_blobs = net_->output_blobs();
vector<Blob<float>*>& input_blobs = net_->input_blobs();
const vector<Blob<float>*>& output_blobs = net_->output_blobs();
const vector<Blob<float>*>& input_blobs = net_->input_blobs();
if (static_cast<unsigned int>(mxGetDimensions(top_diff)[0]) !=
output_blobs.size()) {
mex_error("Invalid input size");