From 5301715c3ea9507e33fd7407551cf1f7ff0d82bf Mon Sep 17 00:00:00 2001 From: Natalia Gimelshein Date: Thu, 24 Mar 2016 10:34:49 -0700 Subject: fix failing SpatialFullConvolution test --- test/test.lua | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index 55171c7..1410b8c 100644 --- a/test/test.lua +++ b/test/test.lua @@ -214,10 +214,10 @@ function cudnntest.SpatialFullConvolution_forward_batch() local kj = math.random(1,15) local si = math.random(1,ki) local sj = math.random(1,kj) - local outi = math.random(1,64) - local outj = math.random(1,64) - local ini = (outi-1)*si+ki - local inj = (outj-1)*sj+kj + local ini = math.random(1,64) + local inj = math.random(1,64) + local outi = (ini-1)*si+ki + local outj = (inj-1)*sj+kj local input = torch.randn(bs,from,inj,ini):cuda() local sconv = nn.SpatialFullConvolution(from,to,ki,kj,si,sj):cuda() @@ -253,10 +253,10 @@ function cudnntest.SpatialFullConvolution_backward_batch() local kj = math.random(1,15) local si = math.random(1,ki) local sj = math.random(1,kj) - local outi = math.random(1,64) - local outj = math.random(1,64) - local ini = (outi-1)*si+ki - local inj = (outj-1)*sj+kj + local ini = math.random(1,64) + local inj = math.random(1,64) + local outi = (ini-1)*si+ki + local outj = (inj-1)*sj+kj local scale = math.random() local input = torch.randn(bs,from,inj,ini):cuda() -- cgit v1.2.3