diff options
author | Soumith Chintala <soumith@gmail.com> | 2015-11-16 20:15:07 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2015-11-16 20:15:07 +0300 |
commit | 0e43c1e96bd3c8cf10158c9bcc32f21775bb7276 (patch) | |
tree | cf19c3157a9dad403bfb9e19b5952e53980872e6 | |
parent | 5fb279988b5648939bb8f5f0c69842b4aa9262c9 (diff) | |
parent | 7a928f775bf2d86978dfed4f7c479634321b33f3 (diff) |
Merge pull request #126 from dm-jrae/master
image test: preserve default tensor type.
-rw-r--r-- | test/test.lua | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/test.lua b/test/test.lua index 5138514..c316edf 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1,5 +1,4 @@ local test = {} -torch.setdefaulttensortype('torch.DoubleTensor') local precision = 1e-4 local precision_mean = 1e-3 local precision_std = 1e-1 @@ -521,11 +520,14 @@ end function image.test(tests, seed) + local defaultTensorType = torch.getdefaulttensortype() + torch.setdefaulttensortype('torch.DoubleTensor') seed = seed or os.time() print('seed: ', seed) math.randomseed(seed) tester = torch.Tester() tester:add(test) tester:run(tests) + torch.setdefaulttensortype(defaultTensorType) return tester end |