diff options
author | Étienne Simon <esimon@esimon.eu> | 2015-07-23 17:52:53 -0400 |
---|---|---|
committer | Étienne Simon <esimon@esimon.eu> | 2015-07-23 17:52:53 -0400 |
commit | e7aba08e6b209ac7f091eb9f08b49a2c90b070ed (patch) | |
tree | f1f684876659db8b77a45ebcb135cb64606ed840 /train.py | |
parent | 1ee965e32c6678ca063d853d80d7bdeae0ab9734 (diff) | |
download | taxi-e7aba08e6b209ac7f091eb9f08b49a2c90b070ed.tar.gz taxi-e7aba08e6b209ac7f091eb9f08b49a2c90b070ed.zip |
Lunch validation every 10000 batches (instead of 1000)
Diffstat (limited to 'train.py')
-rwxr-xr-x | train.py | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -106,16 +106,16 @@ 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=1000), + extensions=[TrainingDataMonitoring(monitored, prefix='train', every_n_batches=10000), DataStreamMonitoring(valid_monitored, valid_stream, prefix='valid', - every_n_batches=1000), + every_n_batches=10000), Printing(every_n_batches=1000), FinishAfter(every_n_batches=10000000), SaveLoadParams(dump_path, cg, before_training=True, # before training -> load params - every_n_batches=1000, # every N batches -> save params + every_n_batches=10000, # every N batches -> save params after_epoch=True, # after epoch -> save params after_training=True, # after training -> save params ), @@ -123,7 +123,7 @@ if __name__ == "__main__": RunOnTest(model_name, model, stream, - every_n_batches=1000), + every_n_batches=10000), ] if use_plot: |