Skip to content

GitLab

Commit d6a1e273 authored by Davis King's avatar Davis King
Browse files

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

parent 9043c741
Hide whitespace changes
Inline Side-by-side
Showing with 38 additions and 0 deletions
dlib/dnn/solvers.h
View file @ d6a1e273
...@@ -89,6 +89,25 @@ namespace dlib ...@@ -89,6 +89,25 @@ namespace dlib
return v; 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 > template < layer_mode mode >
const tensor& operator() ( const tensor& operator() (
const float learning_rate, const float learning_rate,
...@@ -261,6 +280,25 @@ namespace dlib ...@@ -261,6 +280,25 @@ namespace dlib
return s; 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 > template < layer_mode mode >
const tensor& operator() ( const tensor& operator() (
const float learning_rate, 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