diff options
author | Soumith Chintala <soumith@fb.com> | 2015-04-06 02:14:53 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@fb.com> | 2015-04-06 02:14:53 +0300 |
commit | 42366648d1a45fa012f6ded894215898d099d030 (patch) | |
tree | 517745c056a7f97b73583aaad1daeaed9cd16431 /test.lua | |
parent | 326d543cbac6ef45fe41d59edd5f4bf2aae0d9ec (diff) |
reducing adaptive maxpooling's test limits (mostly for travis to not suck)
Diffstat (limited to 'test.lua')
-rw-r--r-- | test.lua | 10 |
1 files changed, 5 insertions, 5 deletions
@@ -1806,10 +1806,10 @@ end function nntest.SpatialAdaptiveMaxPooling() local from = math.random(1,5) - local ki = math.random(1,12) - local kj = math.random(1,12) - local ini = math.random(1,64) - local inj = math.random(1,64) + local ki = math.random(1,5) + local kj = math.random(1,5) + local ini = math.random(1,16) + local inj = math.random(1,16) local module = nn.SpatialAdaptiveMaxPooling(ki,kj) local input = torch.rand(from,ini,inj) @@ -1822,7 +1822,7 @@ function nntest.SpatialAdaptiveMaxPooling() mytester:asserteq(berr, 0, torch.typename(module) .. ' - i/o backward err ') -- batch - local nbatch = math.random(2,5) + local nbatch = math.random(1,3) input = torch.rand(nbatch,from,ini,inj) module = nn.SpatialAdaptiveMaxPooling(ki,kj) |