diff options
author | soumith <soumith@gmail.com> | 2015-08-21 07:06:17 +0300 |
---|---|---|
committer | soumith <soumith@gmail.com> | 2015-08-21 07:06:17 +0300 |
commit | 4c2a0a568a232956f0cbe3c89c5e889df0f0ed94 (patch) | |
tree | 735bf3c04638207e4823e5ad36ec2bb8b27fda62 /SpatialConvolution.lua | |
parent | 96d2b275a345b47a5a2e8f26d1483303321220f8 (diff) |
fix for cudnn r3 perf bug
Diffstat (limited to 'SpatialConvolution.lua')
-rw-r--r-- | SpatialConvolution.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index a99f551..0c1f311 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -20,6 +20,7 @@ function SpatialConvolution:__init(nInputPlane, nOutputPlane, self.gradWeight = torch.Tensor(nOutputPlane, nInputPlane/self.groups, kW, kH) self:reset() self.iSize = torch.LongStorage(4):fill(0) + self.fastest_mode = true end -- if you change the configuration of the module manually, call this |