Fix bug propagating the batch_size correctly. Improves performance on…

… MNIST.
aigamedev · Apr 10, 2015 · 02c8001 · 02c8001
1 parent 2d31592
commit 02c8001
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sknn/mlp.py b/sknn/mlp.py
@@ -141,7 +141,7 @@ def _create_trainer(self, dataset):
 
         return sgd.SGD(
             cost=self.cost,
-            batch_size=1,
+            batch_size=self.batch_size,
             learning_rule=self.learning_rule,
             learning_rate=self.learning_rate,
             termination_criterion=termination_criterion,