diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-07-08 11:08:18 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-07-08 11:08:18 +0300 |
commit | 155d6f59211504e6dc72840133df1eb93a16cfb3 (patch) | |
tree | d34b2bc005ad39aef7042ddecea316ef1f07b73d /SpatialConvolution.lua | |
parent | 3fd281c6e5dcb90bc80ef0083dc778a164c31159 (diff) |
handle half true/pseudo
Diffstat (limited to 'SpatialConvolution.lua')
-rw-r--r-- | SpatialConvolution.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index a224a6d..58c78b2 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -123,7 +123,7 @@ function SpatialConvolution:createIODescriptors(input) errcheck('cudnnSetConvolutionNdDescriptor', self.convDesc[0], 2, pad:data(), stride:data(), upscale:data(), 'CUDNN_CROSS_CORRELATION', - cudnn.configmap[torch.type(self.weight)]); + cudnn.configmap(torch.type(self.weight))); local function destroyConvDesc(d) errcheck('cudnnDestroyConvolutionDescriptor', d[0]); end |