diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2015-03-18 22:28:10 +0300 |
|---|---|---|
| committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2015-03-18 22:29:29 +0300 |
| commit | c1389c1677caf0f8d4902985bb6aa0963147952f (patch) | |
| tree | d7e731d6eededbaca29eb586544067a0638728a8 /test | |
| parent | ae4e4144374623cdf91e75d6c55ae4df91c71ff8 (diff) | |
removed old lines to make avg pooling test pass
Diffstat (limited to 'test')
| -rw-r--r-- | test/test.lua | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/test/test.lua b/test/test.lua index cc634c7..4629697 100644 --- a/test/test.lua +++ b/test/test.lua @@ -334,9 +334,7 @@ function cudnntest.SpatialAveragePooling_batch() local sconv = nn.SpatialAveragePooling(ki,kj,si,sj):cuda() local groundtruth = sconv:forward(input):clone() - groundtruth:mul(1/(ki*kj)) -- difference between nn and cudnn local groundgrad = sconv:backward(input, gradOutput) - groundgrad:mul(1/(ki*kj)) -- difference between nn and cudnn cutorch.synchronize() local gconv = cudnn.SpatialAveragePooling(ki,kj,si,sj):cuda() local rescuda = gconv:forward(input) @@ -369,9 +367,7 @@ function cudnntest.SpatialAveragePooling_single() local sconv = nn.SpatialAveragePooling(ki,kj,si,sj):cuda() local groundtruth = sconv:forward(input):clone() - groundtruth:mul(1/(ki*kj)) -- difference between nn and cudnn local groundgrad = sconv:backward(input, gradOutput) - groundgrad:mul(1/(ki*kj)) -- difference between nn and cudnn cutorch.synchronize() local gconv = cudnn.SpatialAveragePooling(ki,kj,si,sj):cuda() local _ = gconv:forward(input) |