aboutsummaryrefslogtreecommitdiff
path: root/train.py
diff options
context:
space:
mode:
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
commite7aba08e6b209ac7f091eb9f08b49a2c90b070ed (patch)
treef1f684876659db8b77a45ebcb135cb64606ed840 /train.py
parent1ee965e32c6678ca063d853d80d7bdeae0ab9734 (diff)
downloadtaxi-e7aba08e6b209ac7f091eb9f08b49a2c90b070ed.tar.gz
taxi-e7aba08e6b209ac7f091eb9f08b49a2c90b070ed.zip
Lunch validation every 10000 batches (instead of 1000)
Diffstat (limited to 'train.py')
-rwxr-xr-xtrain.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/train.py b/train.py
index 8260218..222f24c 100755
--- a/train.py
+++ b/train.py
@@ -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: