diff options
author | Dominik Grewe <dominikg@google.com> | 2015-05-05 12:29:50 +0300 |
---|---|---|
committer | Dominik Grewe <dominikg@google.com> | 2015-05-05 12:31:27 +0300 |
commit | c5c63d0fd873ff73716ed79e4f7abb9926f4748b (patch) | |
tree | f0e986e1014c125c7b7e30c564f22a0d45ca8d26 /CrossEntropyCriterion.lua | |
parent | e5556492d5efd97cb30a13dab214487a7bc1a6b8 (diff) |
Check for `nn.Module` and `nn.Criterion` in recursiveType.
Diffstat (limited to 'CrossEntropyCriterion.lua')
-rw-r--r-- | CrossEntropyCriterion.lua | 7 |
1 files changed, 0 insertions, 7 deletions
diff --git a/CrossEntropyCriterion.lua b/CrossEntropyCriterion.lua index 2b3c78c..d4d19e5 100644 --- a/CrossEntropyCriterion.lua +++ b/CrossEntropyCriterion.lua @@ -25,11 +25,4 @@ function CrossEntropyCriterion:updateGradInput(input, target) return self.gradInput end -function CrossEntropyCriterion:type(name) - Criterion.type(self, name) - self.lsm:type(name) - self.nll:type(name) - return self -end - return nn.CrossEntropyCriterion |