Commit d6a1e273 authored by Davis King's avatar Davis King

Updated solvers to correctly pull in cont_'s bias parameter multipliers.

parent 9043c741
......@@ -89,6 +89,25 @@ namespace dlib
return v;
}
template <
long _num_filters,
long _nr,
long _nc,
int _stride_y,
int _stride_x,
int _padding_y,
int _padding_x
>
const tensor& operator() (
const float learning_rate,
const cont_<_num_filters,_nr,_nc,_stride_y,_stride_x,_padding_y,_padding_x>& l,
const tensor& params_grad
)
{
update_considering_bias(learning_rate, l, params_grad, params_grad.size()-l.num_filters());
return v;
}
template < layer_mode mode >
const tensor& operator() (
const float learning_rate,
......@@ -261,6 +280,25 @@ namespace dlib
return s;
}
template <
long _num_filters,
long _nr,
long _nc,
int _stride_y,
int _stride_x,
int _padding_y,
int _padding_x
>
const tensor& operator() (
const float learning_rate,
const cont_<_num_filters,_nr,_nc,_stride_y,_stride_x,_padding_y,_padding_x>& l,
const tensor& params_grad
)
{
update_considering_bias(learning_rate, l, params_grad, params_grad.size()-l.num_filters());
return s;
}
template < layer_mode mode >
const tensor& operator() (
const float learning_rate,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment