diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-03-17 01:09:41 +0300 |
|---|---|---|
| committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-03-17 01:09:41 +0300 |
| commit | 6765359c156615f482612e3eb4b3655623c76a8c (patch) | |
| tree | 4ad58de8e453bba3825c50fb6fc69ae4c4f1476e /convert.lua | |
| parent | 27254fd4787aa9e1d126215fa7a98421219c9f66 (diff) | |
fix cudnn -> nn avg-pooling conversion
Diffstat (limited to 'convert.lua')
| -rw-r--r-- | convert.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/convert.lua b/convert.lua index 49ea2b3..7b51ae5 100644 --- a/convert.lua +++ b/convert.lua @@ -42,6 +42,7 @@ function cudnn.convert(net, dst) if v == 'ReLU' then y = dst.ReLU() end -- because parameters for k,u in pairs(x) do y[k] = u end if src == cudnn and x.clearDesc then x:clearDesc() end + if src == cudnn and v == 'SpatialAveragePooling' then y.divide = true end return y end local t = torch.typename(x) |