From a1d92fb92eb4f97373ee677a0650a23b8d530bd8 Mon Sep 17 00:00:00 2001 From: AntreasAntoniou Date: Tue, 7 Nov 2017 12:55:23 +0000 Subject: [PATCH] Add evaluation phase for stochastic layers --- mlp/optimisers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlp/optimisers.py b/mlp/optimisers.py index 629144b..8ab313a 100644 --- a/mlp/optimisers.py +++ b/mlp/optimisers.py @@ -80,7 +80,7 @@ class Optimiser(object): data_mon_vals = OrderedDict([(key + label, 0.) for key in self.data_monitors.keys()]) for inputs_batch, targets_batch in dataset: - activations = self.model.fprop(inputs_batch) + activations = self.model.fprop(inputs_batch, evaluation=True) for key, data_monitor in self.data_monitors.items(): data_mon_vals[key + label] += data_monitor( activations[-1], targets_batch)