diff options
author | soumith <soumith@fb.com> | 2015-01-02 23:07:32 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-01-02 23:07:32 +0300 |
commit | df5925a2f094faf0d3cfd804cd7bea1617aab7df (patch) | |
tree | 07b47a5f8eca80230cdc713ee912c5e955de86bf /ReLU.lua | |
parent | e2595a3abfaea264ccbae70557ea3013961927be (diff) |
adding inplace non-linearities
Diffstat (limited to 'ReLU.lua')
-rw-r--r-- | ReLU.lua | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -1,6 +1,6 @@ local ReLU, parent = torch.class('cudnn.ReLU','cudnn._Pointwise') -function ReLU:__init() - parent.__init(self) +function ReLU:__init(inplace) + parent.__init(self, inplace) self.mode = 'CUDNN_ACTIVATION_RELU' end |