From ab0ee1fe41b9843dc2469541b3f0ff856c12e547 Mon Sep 17 00:00:00 2001 From: Nicholas Leonard Date: Thu, 3 Aug 2017 15:22:55 -0400 Subject: BN supports batchsize=1 --- test.lua | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test.lua b/test.lua index fb65bd9..02b63d4 100644 --- a/test.lua +++ b/test.lua @@ -978,9 +978,9 @@ local function BatchNormalization_backward(moduleName, mode, inputSize, backward end end -local function testBatchNormalization(name, dim, k) +local function testBatchNormalization(name, dim, k, batchsize) local function inputSize() - local inputSize = { torch.random(2,32), torch.random(1, k) } + local inputSize = { batchsize or torch.random(2,32), torch.random(1, k) } for i=1,dim do table.insert(inputSize, torch.random(1,k)) end @@ -1005,6 +1005,7 @@ end function cunntest.BatchNormalization() testBatchNormalization('BatchNormalization', 0, 128) + testBatchNormalization('BatchNormalization', 0, 128, 1) -- test batchsize=1 end function cunntest.SpatialBatchNormalization() -- cgit v1.2.3