commit 72e1c4ed6aceb97bf016c84df82cbfadd9d4612d
parent d20bc216b29903a6fa9e7a98b79349b02b4f5a2e
Author: Étienne Simon <esimon@esimon.eu>
Date: Mon, 26 May 2014 16:44:58 +0200
Use sum instead of mean over mini-batches
Diffstat:
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/model.py b/model.py
@@ -81,8 +81,8 @@ class Model(object):
violating_margin_left = score_left>0
violating_margin_right = score_right>0
- criterion_left = T.mean(violating_margin_left*score_left)
- criterion_right = T.mean(violating_margin_right*score_right)
+ criterion_left = T.sum(violating_margin_left*score_left)
+ criterion_right = T.sum(violating_margin_right*score_right)
criterion = criterion_left + criterion_right
self.train_function = theano.function(inputs=inputs, outputs=[criterion], updates=self.updates(criterion))