From 7c15286b6dadd1adc1f7406faed402a4bfe770f3 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Tue, 28 Jul 2015 09:48:31 -0400 Subject: Memory net changes --- train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 1d59bb5..27d9b77 100755 --- a/train.py +++ b/train.py @@ -114,14 +114,15 @@ if __name__ == "__main__": extensions=[TrainingDataMonitoring(monitored, prefix='train', every_n_batches=monitor_freq), DataStreamMonitoring(valid_monitored, valid_stream, prefix='valid', - every_n_batches=monitor_freq), + every_n_batches=monitor_freq, + after_epoch=False), Printing(every_n_batches=monitor_freq), FinishAfter(every_n_batches=10000000), SaveLoadParams(dump_path, cg, before_training=True, # before training -> load params every_n_batches=monitor_freq,# every N batches -> save params - after_epoch=True, # after epoch -> save params + after_epoch=False, after_training=True, # after training -> save params ), -- cgit v1.2.3