diff options
author | Georg Ostrovski <ostrovski@google.com> | 2016-03-06 05:33:33 +0300 |
---|---|---|
committer | Georg Ostrovski <ostrovski@google.com> | 2016-03-06 05:33:33 +0300 |
commit | b26cafc41a8d944a838b8ff87059b5c6ff32d1ea (patch) | |
tree | 473cc04db00061807b293e3d8717c91ac8847d97 | |
parent | 96dcfbf0bd18ba8e8cc5afdf4e8fb05417f1d86a (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.lua | 4 |
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) |