OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

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

of the parameter vectors.
This commit is contained in:
Davis King 2016-06-05 16:39:10 -04:00
parent 3c002d1cff
commit aa19ff7377
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dlib/dnn/solvers.h
View File

@ -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;
}