Add evaluation phase for stochastic layers

2017-11-07 12:55:23 +00:00 · 2017-11-07 12:55:23 +00:00 · a1d92fb92e
commit a1d92fb92e
parent 1021afa640
1 changed files with 1 additions and 1 deletions
--- 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)