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