diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-08-03 06:58:29 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-08-03 06:58:29 +0300 |
commit | b812efdd1c3465547f0e5b0e24d2053f95ebfe2e (patch) | |
tree | 5217fb65540afbdbad4d72fa0abf9bff28ade3d3 | |
parent | a50b2f4061e5014b24b70b8ffa6db0136f956b17 (diff) | |
parent | 3f36de1c2859be04b64d85ee56153afe1c01041e (diff) |
Merge pull request #230 from shuzi/master
adding supporting function to allow TemporalConvolution to have dermiā¦
-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 |