diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-03-24 21:33:44 +0300 |
|---|---|---|
| committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-03-24 21:33:44 +0300 |
| commit | fff9138852530904df3410dbba9fd0c940b7b3f1 (patch) | |
| tree | a39585d0cf2eba211f039b2756c648223b29347b /convert.lua | |
| parent | 71181331598e191cf805998fa55742fe5bcedd19 (diff) | |
fix conversion for saved cudnn->nn
Diffstat (limited to 'convert.lua')
| -rw-r--r-- | convert.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/convert.lua b/convert.lua index aa8c65f..a938ddc 100644 --- a/convert.lua +++ b/convert.lua @@ -42,7 +42,7 @@ function cudnn.convert(net, dst) torch.setmetatable(y, dst_prefix..v) 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 x.clearDesc then x.clearDesc(y) end if src == cudnn and v == 'SpatialAveragePooling' then y.divide = true end return y end |