diff options
author | Anthony Sandrin <asandrin@nvidia.com> | 2016-03-17 00:29:08 +0300 |
---|---|---|
committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-04-18 23:18:43 +0300 |
commit | 60a66872730eceaf4f769c281e2ad7289272323e (patch) | |
tree | 732547d0054234a07f57f534642b27423267ebb0 /test | |
parent | be65e83fb6d0aaa8b2842c9f584b3d85abcff011 (diff) |
Add utility functions and correct tensor dimensions.
Tensor dimensions needed to be reversed since RNN function expect stride 1
in the first dimension.
Remove experimental RNN scripts.
Make inputSize a parameter to the RNN layer and use strings for enum values.
* add missing cutorch.sychronize() to SpatialConvolution benchmark
* rename bench to benchSpatial
* add benchVolumetric
Set weights in __init()
Remove unnecessary input argument to resetIODescriptors
- added new convolution modes
- cleaned up benchVolumetric
Add missing synchronization
test reverted to non-verbose and random optimizer
Return output from RNN:updateOutput and zero cell output.
cuDNN does not set cell output for RNN_RELU so the output was garbage.
Check for nil cx/hx when not training.
Correct self.RNNDesc to self.rnnDesc
Diffstat (limited to 'test')
-rw-r--r-- | test/test.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test.lua b/test/test.lua index 8fcd1b9..ba1b5a1 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1444,10 +1444,10 @@ math.randomseed(os.time()) mytester = torch.Tester() mytester:add(cudnntest) --- if torch.random(1,2) == 1 then +if torch.random(1,2) == 1 then cudnn.benchmark = true -- run manual auto-tuner - cudnn.verbose = true --- end +-- cudnn.verbose = true +end for i=1,cutorch.getDeviceCount() do |