diff options
author | Nicholas Leonard <nick@nikopia.org> | 2015-07-28 18:14:25 +0300 |
---|---|---|
committer | Nicholas Leonard <nick@nikopia.org> | 2015-07-28 18:14:25 +0300 |
commit | a9026fd1377757a148c464784e2d2dce6580c138 (patch) | |
tree | 5e4b98ce665a3afd460467406dfff28be2c2d14b | |
parent | 8fb2f5deae1d805e375bb2ba514e6b5452aca3ad (diff) |
TreeNLLCriterion._module support in cunn
-rw-r--r-- | TreeNLLCriterion.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/TreeNLLCriterion.lua b/TreeNLLCriterion.lua index b8e1d04..02ea2b2 100644 --- a/TreeNLLCriterion.lua +++ b/TreeNLLCriterion.lua @@ -10,7 +10,7 @@ local TreeNLLCriterion, parent = torch.class("nn.TreeNLLCriterion", "nn.Criterion") function TreeNLLCriterion:__init() - self._module = nn.Mean() --not in cunn + self._module = nn.Mean() parent.__init(self) self._output_grad = torch.Tensor{-1} end |