Use less memory in VolumetricDilatedMaxPooling tests

Previously, VolumetricDilatedMaxPooling_backward_batch could run out of
memory even on a 12GB GPU.

1 files changed, 7 insertions, 7 deletions

diff --git a/test.lua b/test.lua
index 598198f..06cdbb6 100644
--- a/test.lua
+++ b/test.lua
@@ -4469,8 +4469,8 @@ function cunntest.VolumetricMaxPooling_backward()
 end
 
 function cunntest.VolumetricDilatedMaxPooling_forward_batch()
-   local bs = math.random(8,16)
-   local from = math.random(8,16)
+   local bs = math.random(4,8)
+   local from = math.random(4,8)
    local to = from
    local kt = math.random(2,4)
    local ki = math.random(2,4)
@@ -4513,8 +4513,8 @@ function cunntest.VolumetricDilatedMaxPooling_forward_batch()
 end
 
 function cunntest.VolumetricDilatedMaxPooling_backward_batch()
-   local bs = math.random(8,16)
-   local from = math.random(8,16)
+   local bs = math.random(4,8)
+   local from = math.random(4,8)
    local to = from
    local kt = math.random(2,4)
    local ki = math.random(2,4)
@@ -4522,9 +4522,9 @@ function cunntest.VolumetricDilatedMaxPooling_backward_batch()
    local st = math.random(2,4)
    local si = math.random(2,4)
    local sj = math.random(2,4)
-   local outt = math.random(32,60)
-   local outi = math.random(32,60)
-   local outj = math.random(32,60)
+   local outt = math.random(8,16)
+   local outi = math.random(8,16)
+   local outj = math.random(8,16)
    local padt = math.random(0,kt/2-1)
    local padi = math.random(0,ki/2-1)
    local padj = math.random(0,kj/2-1)