From 88cdc3f8047a05bc5971eaa915ca6626f89a3e78 Mon Sep 17 00:00:00 2001 From: AdeB Date: Wed, 24 Jun 2015 15:12:15 -0400 Subject: New configs. training step rule out of train.py --- train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 1b75138..83317c9 100755 --- a/train.py +++ b/train.py @@ -11,7 +11,7 @@ from functools import reduce from theano import tensor from blocks import roles -from blocks.algorithms import AdaDelta, CompositeRule, GradientDescent, RemoveNotFinite, StepRule +from blocks.algorithms import AdaDelta, CompositeRule, GradientDescent, RemoveNotFinite, StepRule, Momentum from blocks.extensions import Printing, FinishAfter, SimpleExtension from blocks.extensions.monitoring import DataStreamMonitoring, TrainingDataMonitoring @@ -136,8 +136,7 @@ if __name__ == "__main__": cost=cost, step_rule=CompositeRule([ ElementwiseRemoveNotFinite(), - AdaDelta(), - #Momentum(learning_rate=config.learning_rate, momentum=config.momentum), + config.step_rule, ]), params=params) -- cgit v1.2.3