diff options
author | Natalia Gimelshein <ngimelshein@nvidia.com> | 2016-03-24 20:34:49 +0300 |
---|---|---|
committer | Natalia Gimelshein <ngimelshein@nvidia.com> | 2016-03-24 21:22:03 +0300 |
commit | 5301715c3ea9507e33fd7407551cf1f7ff0d82bf (patch) | |
tree | e02a866bf528a0854237d5d5c96e05bca38c962e | |
parent | 7e74cb7464106114b468289f5dcc59744b2d25cd (diff) |
fix failing SpatialFullConvolution test
-rw-r--r-- | test/test.lua | 16 |
1 files changed, 8 insertions, 8 deletions
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() |