commit e7aba08e6b209ac7f091eb9f08b49a2c90b070ed
parent 1ee965e32c6678ca063d853d80d7bdeae0ab9734
Author: Étienne Simon <esimon@esimon.eu>
Date: Thu, 23 Jul 2015 17:52:53 -0400
Lunch validation every 10000 batches (instead of 1000)
Diffstat:
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git 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: