Add monitor_freq config variable

1 files changed, 10 insertions, 5 deletions

diff --git a/train.py b/train.py
index b4ef774..1d59bb5 100755
--- a/train.py
+++ b/train.py
@@ -106,16 +106,21 @@ if __name__ == "__main__":
     dump_path = os.path.join('model_data', model_name) + '.pkl'
     logger.info('Dump path: %s' % dump_path)
 
-    extensions=[TrainingDataMonitoring(monitored, prefix='train', every_n_batches=10000),
+    if hasattr(config, 'monitor_freq'):
+        monitor_freq = config.monitor_freq
+    else:
+        monitor_freq = 10000
+
+    extensions=[TrainingDataMonitoring(monitored, prefix='train', every_n_batches=monitor_freq),
                 DataStreamMonitoring(valid_monitored, valid_stream,
                                      prefix='valid',
-                                     every_n_batches=10000),
-                Printing(every_n_batches=10000),
+                                     every_n_batches=monitor_freq),
+                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=10000,       # every N batches -> save params
+                               every_n_batches=monitor_freq,# every N batches -> save params
                                after_epoch=True,            # after epoch -> save params
                                after_training=True,         # after training -> save params
                                ),
@@ -123,7 +128,7 @@ if __name__ == "__main__":
                 RunOnTest(model_name,
                           model,
                           stream,
-                          every_n_batches=10000),
+                          every_n_batches=monitor_freq),
                 ]
 
     if '--progress' in sys.argv: