From 9a779f7328a712a20dd393bdf32c6a84bf9fbe52 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Thu, 21 May 2015 10:46:05 -0400 Subject: Model changes --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 96dd798..677ed45 100755 --- a/train.py +++ b/train.py @@ -73,7 +73,7 @@ def setup_test_stream(req_vars): test = transformers.TaxiAddFirstLastLen(config.n_begin_end_pts, test) test = transformers.Select(test, tuple(req_vars)) - test_stream = Batch(test, iteration_scheme=ConstantScheme(1000)) + test_stream = Batch(test, iteration_scheme=ConstantScheme(1)) return test_stream @@ -100,8 +100,8 @@ def main(): cost=cost, step_rule=CompositeRule([ RemoveNotFinite(), - #AdaDelta(decay_rate=0.95), - Momentum(learning_rate=config.learning_rate, momentum=config.momentum), + AdaDelta(), + #Momentum(learning_rate=config.learning_rate, momentum=config.momentum), ]), params=params) -- cgit v1.2.3