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
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;
}
......@@ -85,7 +85,7 @@ namespace dlib
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;
}
......@@ -232,7 +232,7 @@ namespace dlib
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;
}
......@@ -251,7 +251,7 @@ namespace dlib
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;
}
......
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