From 52cf8293061a1e35b5b443ef6dc70aa51727cf00 Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Mon, 30 Nov 2015 16:45:25 -0500 Subject: syncing with Thibaut --- simulation/vi_blocks.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'simulation/vi_blocks.py') diff --git a/simulation/vi_blocks.py b/simulation/vi_blocks.py index dcf6b46..2b03198 100644 --- a/simulation/vi_blocks.py +++ b/simulation/vi_blocks.py @@ -67,16 +67,17 @@ if __name__ == "__main__": alg = blocks.algorithms.GradientDescent(cost=cost, parameters=[mu, sig], step_rule=step_rules) - #data_stream = ab.create_fixed_data_stream(n_cascades, graph, batch_size, - # shuffle=False) - data_stream = ab.create_learned_data_stream(graph, batch_size) + data_stream = ab.create_fixed_data_stream(n_cascades, graph, batch_size, + shuffle=False) + #data_stream = ab.create_learned_data_stream(graph, batch_size) loop = blocks.main_loop.MainLoop( alg, data_stream, extensions=[ blocks.extensions.FinishAfter(after_n_batches = 10**4), blocks.extensions.monitoring.TrainingDataMonitoring([cost, mu, sig, rmse, g_shared], after_batch=True), - blocks.extensions.Printing(every_n_batches = 100), + blocks.extensions.Printing(every_n_batches = 100, + after_epoch=False), ] ) loop.run() -- cgit v1.2.3-70-g09d2