diff options
| author | nicholas-leonard <nick@nikopia.org> | 2015-04-20 18:47:39 +0300 |
|---|---|---|
| committer | nicholas-leonard <nick@nikopia.org> | 2015-04-20 18:47:39 +0300 |
| commit | 35cfc17ee40ceae4916b1f1754fbdb5eea10c5aa (patch) | |
| tree | 08482cbafb7eb98c398982e792f2be0393ea5904 /SoftMaxTree.lua | |
| parent | b10c3aa2b51b3c6c848c4d8d2c2517ac59872269 (diff) | |
SoftMaxTree:parameters() returns scales and size
Diffstat (limited to 'SoftMaxTree.lua')
| -rw-r--r-- | SoftMaxTree.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/SoftMaxTree.lua b/SoftMaxTree.lua index e3ff61e..af4be64 100644 --- a/SoftMaxTree.lua +++ b/SoftMaxTree.lua @@ -232,7 +232,7 @@ function SoftMaxTree:parameters() return {self.weight, self.bias}, {self.gradWeight, self.gradBias} end end - return params, grads + return params, grads, {}, self.nChildNode*2 end function SoftMaxTree:updateParameters(learningRate) |