diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-11-11 19:38:11 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-11-11 19:38:11 +0400 |
commit | 98fb40a126afb2f33800af2f081d3119a8b55448 (patch) | |
tree | b811fd62ac82766bc9d1a49f4d9e99cdc52ec8fa /SparseCriterion.lua | |
parent | a8ab27537e16f4440565fbebe6313b2172b489a1 (diff) |
Upgraded all code to new torch master.torch-master
Diffstat (limited to 'SparseCriterion.lua')
-rw-r--r-- | SparseCriterion.lua | 18 |
1 files changed, 4 insertions, 14 deletions
diff --git a/SparseCriterion.lua b/SparseCriterion.lua index ddaa75c..d086f28 100644 --- a/SparseCriterion.lua +++ b/SparseCriterion.lua @@ -5,22 +5,12 @@ function SparseCriterion:__init() self.sizeAverage = true end -function SparseCriterion:forward(input) - input.nn.SparseCriterion_forward(self, input) +function SparseCriterion:updateOutput(input) + input.nn.SparseCriterion_updateOutput(self, input) return self.output end -function SparseCriterion:backward(input) - input.nn.SparseCriterion_backward(self, input) +function SparseCriterion:updateGradInput(input) + input.nn.SparseCriterion_updateGradInput(self, input) return self.gradInput end - -function SparseCriterion:write(file) - parent.write(self, file) - file:writeBool(self.sizeAverage) -end - -function SparseCriterion:read(file) - parent.read(self, file) - self.sizeAverage = file:readBool() -end |