From f4d3ee6449217535bdbe19ac9c5fdd825d71b0d3 Mon Sep 17 00:00:00 2001 From: AdeB Date: Tue, 5 May 2015 22:15:22 -0400 Subject: New hyperparameters. Training error is monitored. --- train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 4cbd526..f501cba 100644 --- a/train.py +++ b/train.py @@ -27,7 +27,7 @@ from blocks.graph import ComputationGraph from blocks.main_loop import MainLoop from blocks.extensions import Printing, FinishAfter from blocks.extensions.saveload import Dump, LoadFromDump, Checkpoint -from blocks.extensions.monitoring import DataStreamMonitoring +from blocks.extensions.monitoring import DataStreamMonitoring, TrainingDataMonitoring import data import transformers @@ -107,7 +107,8 @@ def main(): step_rule=Momentum(learning_rate=config.learning_rate, momentum=config.momentum), params=params) - extensions=[DataStreamMonitoring(model.monitor, valid_stream, + extensions=[TrainingDataMonitoring(model.monitor, prefix='train', every_n_batches=1000), + DataStreamMonitoring(model.monitor, valid_stream, prefix='valid', every_n_batches=1000), Printing(every_n_batches=1000), -- cgit v1.2.3