diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_rnn.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_rnn.lua b/test/test_rnn.lua index e7ee3de..2476ce4 100644 --- a/test/test_rnn.lua +++ b/test/test_rnn.lua @@ -216,7 +216,7 @@ function getRNNCheckSums(miniBatch, seqLength, hiddenSize, numberOfLayers, numbe cudnn.getHandle(), rnn.rnnDesc[0], layer, - rnn.xDescs, + rnn.xDescs[0], rnn.wDesc[0], rnn.weight:data(), layerId, @@ -247,7 +247,7 @@ function getRNNCheckSums(miniBatch, seqLength, hiddenSize, numberOfLayers, numbe cudnn.getHandle(), rnn.rnnDesc[0], layer, - rnn.xDescs, + rnn.xDescs[0], rnn.wDesc[0], rnn.weight:data(), layerId, @@ -313,4 +313,4 @@ end mytester = torch.Tester() mytester:add(cudnntest) -mytester:run()
\ No newline at end of file +mytester:run() |