diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-04-14 02:17:09 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-04-14 02:18:23 +0300 |
commit | 89929ef0bbf60896134e20ae9a20a143529a7543 (patch) | |
tree | 6d72cbffe9693cd2a2f97c1e239a857895330cda /test | |
parent | 70adafcae259f67129c2de6e1048594aa0283e59 (diff) |
fix running_var meaning in BN
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) |