diff options
| author | Nicholas Leonard <nick@nikopia.org> | 2015-04-27 19:00:21 +0300 |
|---|---|---|
| committer | Nicholas Leonard <nick@nikopia.org> | 2015-04-27 19:00:21 +0300 |
| commit | 955e4960577f8d277a50d484cdc953a5aabef6b1 (patch) | |
| tree | a7346d4f6c8252f32f16337fae03827672824f3c /MarginRankingCriterion.lua | |
| parent | cedc0b997fe39751726632622f52ccc547e85395 (diff) | |
MarginRankingCriterion:type()
Diffstat (limited to 'MarginRankingCriterion.lua')
| -rw-r--r-- | MarginRankingCriterion.lua | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/MarginRankingCriterion.lua b/MarginRankingCriterion.lua index 30c6855..c0573ac 100644 --- a/MarginRankingCriterion.lua +++ b/MarginRankingCriterion.lua @@ -69,3 +69,9 @@ function MarginRankingCriterion:updateGradInput(input, y) end return self.gradInput end + +function MarginRankingCriterion:type(type) + self.gradInput[1] = self.gradInput[1]:type(type) + self.gradInput[2] = self.gradInput[2]:type(type) + return parent.type(self, type) +end |