From 16e5e41530ff5e399b13fd7d36a0a2bffff0454a Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Wed, 8 Jun 2016 11:52:43 -0700 Subject: Natalia fix corner case where both reference impl and cudnn produce nans --- test/test.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index 3d2521e..a448317 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1291,7 +1291,7 @@ end function cudnntest.BatchNormalization() local size = { - math.random(1, 32), + math.random(2, 32), math.random(16, 256), } testBatchNormalization('BatchNormalization', size) -- cgit v1.2.3