From 9313c8bb7f62b14f8c15d2119ae678641f751dbd Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Thu, 23 Jul 2015 10:13:28 -0400 Subject: Can do test with batches of size >1 --- model/mlp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model/mlp.py') diff --git a/model/mlp.py b/model/mlp.py index 7d04c82..d24b2cc 100644 --- a/model/mlp.py +++ b/model/mlp.py @@ -106,7 +106,7 @@ class Stream(object): stream = transformers.taxi_add_first_last_len(stream, self.config.n_begin_end_pts) stream = transformers.taxi_remove_test_only_clients(stream) - return Batch(stream, iteration_scheme=ConstantScheme(1)) + return Batch(stream, iteration_scheme=ConstantScheme(self.config.batch_size)) def inputs(self): return {'call_type': tensor.bvector('call_type'), -- cgit v1.2.3