Commit aa19ff73 authored by Davis King's avatar Davis King

Fixed solvers so they apply the bias specific multipliers to the correct parts

of the parameter vectors.
parent 3c002d1c
...@@ -66,7 +66,7 @@ namespace dlib ...@@ -66,7 +66,7 @@ namespace dlib
const tensor& params_grad const tensor& params_grad
) )
{ {
update_considering_bias(learning_rate, l, params_grad, l.get_num_outputs()); update_considering_bias(learning_rate, l, params_grad, params_grad.size()-l.get_num_outputs());
return v; return v;
} }
...@@ -85,7 +85,7 @@ namespace dlib ...@@ -85,7 +85,7 @@ namespace dlib
const tensor& params_grad const tensor& params_grad
) )
{ {
update_considering_bias(learning_rate, l, params_grad, l.num_filters()); update_considering_bias(learning_rate, l, params_grad, params_grad.size()-l.num_filters());
return v; return v;
} }
...@@ -232,7 +232,7 @@ namespace dlib ...@@ -232,7 +232,7 @@ namespace dlib
const tensor& params_grad const tensor& params_grad
) )
{ {
update_considering_bias(learning_rate, l, params_grad, l.get_num_outputs()); update_considering_bias(learning_rate, l, params_grad, params_grad.size()-l.get_num_outputs());
return s; return s;
} }
...@@ -251,7 +251,7 @@ namespace dlib ...@@ -251,7 +251,7 @@ namespace dlib
const tensor& params_grad const tensor& params_grad
) )
{ {
update_considering_bias(learning_rate, l, params_grad, l.num_filters()); update_considering_bias(learning_rate, l, params_grad, params_grad.size()-l.num_filters());
return s; return s;
} }
......
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