diff options
author | Minwei Feng <fengminwei@gmail.com> | 2016-08-03 06:54:39 +0300 |
---|---|---|
committer | Minwei Feng <fengminwei@gmail.com> | 2016-08-03 06:54:39 +0300 |
commit | 3f36de1c2859be04b64d85ee56153afe1c01041e (patch) | |
tree | 5217fb65540afbdbad4d72fa0abf9bff28ade3d3 | |
parent | a50b2f4061e5014b24b70b8ffa6db0136f956b17 (diff) |
adding supporting function to allow TemporalConvolution to have derministic convolution
-rw-r--r-- | TemporalConvolution.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/TemporalConvolution.lua b/TemporalConvolution.lua index cc9e079..4648ffd 100644 --- a/TemporalConvolution.lua +++ b/TemporalConvolution.lua @@ -39,6 +39,11 @@ function TemporalConvolution:fastest(mode) return self end +function TemporalConvolution:setMode(fmode, bdmode, bwmode) + self = cudnn.SpatialConvolution.setMode(self,fmode, bdmode, bwmode) + return self +end + function TemporalConvolution:resetWeightDescriptors() cudnn.SpatialConvolution.resetWeightDescriptors(self) end |