diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-01-13 16:48:24 +0300 |
|---|---|---|
| committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-01-13 16:48:24 +0300 |
| commit | 9bc3cbac4f054438f5e77824f868cd94e9e22f81 (patch) | |
| tree | 5f74b92207d75fb0ac7865e09d05c26c68666a13 | |
| parent | 4fdde0a10e3240959a5924e2f0d0be495a467b1f (diff) | |
| parent | f13fb810fe918f2e0954177adf346ba3cef7e712 (diff) | |
Merge pull request #90 from Moodstocks/maxpl_test
Fix typo in SpatialMaxPooling test
| -rw-r--r-- | test/test.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test.lua b/test/test.lua index 745da59..7eeb0e8 100644 --- a/test/test.lua +++ b/test/test.lua @@ -366,7 +366,7 @@ function cudnntest.SpatialMaxPooling_batch() local groundgrad = sconv:backward(input, gradOutput) cutorch.synchronize() local gconv = cudnn.SpatialMaxPooling(ki,kj,si,sj,padi,padj):cuda() - if ceil_mode then sconv:ceil() end + if ceil_mode then gconv:ceil() end local rescuda = gconv:forward(input) -- serialize and deserialize torch.save('modelTemp.t7', gconv) @@ -409,7 +409,7 @@ function cudnntest.SpatialMaxPooling_single() local groundgrad = sconv:backward(input, gradOutput) cutorch.synchronize() local gconv = cudnn.SpatialMaxPooling(ki,kj,si,sj,padi,padj):cuda() - if ceil_mode then sconv:ceil() end + if ceil_mode then gconv:ceil() end local _ = gconv:forward(input) -- serialize and deserialize torch.save('modelTemp.t7', gconv) |