From 6cf80aff3e4dc57d13a33c2946bc0ae57cfae6b8 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Thu, 2 Jul 2015 14:32:23 -0400 Subject: More determinism --- train.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 77dca53..e3244ec 100755 --- a/train.py +++ b/train.py @@ -87,6 +87,9 @@ if __name__ == "__main__": else: step_rule = AdaDelta() + logger.info("Fuel seed: %d" % fuel.config.default_seed) + logger.info("Blocks seed: %d" % blocks.config.default_seed) + params = cg.parameters algorithm = GradientDescent( cost=cost, @@ -107,13 +110,14 @@ if __name__ == "__main__": prefix='valid', every_n_batches=1000), Printing(every_n_batches=1000), - - # SaveLoadParams(dump_path, cg, - # before_training=True, # before training -> load params - # every_n_batches=1000, # every N batches -> save params - # after_epoch=True, # after epoch -> save params - # after_training=True, # after training -> save params - # ), + # FinishAfter(every_n_batches=10), + + SaveLoadParams(dump_path, cg, + before_training=True, # before training -> load params + every_n_batches=1000, # every N batches -> save params + after_epoch=True, # after epoch -> save params + after_training=True, # after training -> save params + ), RunOnTest(model_name, model, -- cgit v1.2.3