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 /LSTM.lua | |
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 'LSTM.lua')
0 files changed, 0 insertions, 0 deletions