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

Added bn layer tests.

This commit is contained in:
Davis King 2015-12-07 21:43:39 -05:00
parent 7550681b3c
commit 79344339df
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
dlib/test/dnn.cpp
View File

@ -203,8 +203,8 @@ namespace
gamma_grad.copy_size(gamma);
beta_grad.copy_size(beta);
src_grad = 0;
gamma_grad = 0;
beta_grad = 0;
gamma_grad = 8;
beta_grad = 8;
batch_normalize_gradient bng;
bng(gradient_input, means, vars, src, gamma, src_grad, gamma_grad, beta_grad);
@ -280,8 +280,8 @@ namespace
gamma_grad.copy_size(gamma);
beta_grad.copy_size(beta);
src_grad = 0;
gamma_grad = 0;
beta_grad = 0;
gamma_grad = 9;
beta_grad = 9;
batch_normalize_conv_gradient bng;
bng(gradient_input, means, vars, src, gamma, src_grad, gamma_grad, beta_grad);
@ -566,6 +566,11 @@ namespace
void test_layers()
{
{
print_spinner();
bn_ l;
DLIB_TEST_MSG(test_layer(l), test_layer(l));
}
{
print_spinner();
con_ l(3,3,3,2,2);