diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/test.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test.lua b/test/test.lua index e2ac33c..b7e5376 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1242,7 +1242,7 @@ local function testBatchNormalization(moduleName, inputSize) cbn.weight:copy(gbn.weight) cbn.bias:copy(gbn.bias) mytester:asserteq(cbn.running_mean:mean(), 0, 'error on BN running_mean init') - mytester:asserteq(cbn.running_std:mean(), 1, 'error on BN running_std init') + mytester:asserteq(cbn.running_var:mean(), 1, 'error on BN running_var init') local rescuda = cbn:forward(input) local groundtruth = gbn:forward(input) local resgrad = cbn:backward(input, gradOutput) |