Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/clementfarabet/lua---nnx.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorg Ostrovski <ostrovski@google.com>2016-03-06 05:33:33 +0300
committerGeorg Ostrovski <ostrovski@google.com>2016-03-06 05:33:33 +0300
commitb26cafc41a8d944a838b8ff87059b5c6ff32d1ea (patch)
tree473cc04db00061807b293e3d8717c91ac8847d97
parent96dcfbf0bd18ba8e8cc5afdf4e8fb05417f1d86a (diff)
Change maximal tensor size in test
With the previous setting, the output size of the module could be quite large, and the Jacobian test below (which uses O(N*M) memory for N, M the module in/output size) could run out of memory (taking up to several GB).
-rw-r--r--test/test-all.lua4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test-all.lua b/test/test-all.lua
index 27efc37..c1ca354 100644
--- a/test/test-all.lua
+++ b/test/test-all.lua
@@ -89,8 +89,8 @@ local function template_SpatialReSamplingEx(up, mode)
end
local xratio, yratio
if up then
- xratio = torch.uniform(1.5, 10)
- yratio = torch.uniform(1.5, 10)
+ xratio = torch.uniform(1.5, 5)
+ yratio = torch.uniform(1.5, 5)
else
xratio = torch.uniform(0.41, 0.7)
yratio = torch.uniform(0.41, 0.7)