diff options
author | Nimalan Mahendran <nimalan@twitter.com> | 2016-02-16 23:09:14 +0300 |
---|---|---|
committer | Nimalan Mahendran <nimalan@twitter.com> | 2016-02-26 22:42:46 +0300 |
commit | e7a519d599d056576d7b3801795bc3365d202925 (patch) | |
tree | 48f87aef11041336a9fc64ec2a08e4cd4c3083ae /THNN.lua | |
parent | 6f9543d2fc0a8fd2c15a420ee0f5fc7e0e341ff9 (diff) |
Making margin parameterizable in nn.MultiMarginCriterion
Diffstat (limited to 'THNN.lua')
-rw-r--r-- | THNN.lua | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -202,7 +202,8 @@ TH_API void THNN_(MultiMarginCriterion_updateOutput)( THTensor *output, bool sizeAverage, int p, - THTensor *weights); + THTensor *weights, + real margin); TH_API void THNN_(MultiMarginCriterion_updateGradInput)( THNNState *state, THTensor *input, @@ -210,7 +211,8 @@ TH_API void THNN_(MultiMarginCriterion_updateGradInput)( THTensor *gradInput, bool sizeAverage, int p, - THTensor *weights); + THTensor *weights, + real margin); TH_API void THNN_(PReLU_updateOutput)( THNNState *state, |