Fixing use of global rng in DropoutLayer fprop.

2016-11-09 19:55:11 +00:00 · 2016-11-09 19:55:11 +00:00 · 77a96cfc01
commit 77a96cfc01
parent 9c86d00078
1 changed files with 1 additions and 1 deletions
--- a/mlp/layers.py
+++ b/mlp/layers.py
@ -577,7 +577,7 @@ class DropoutLayer(StochasticLayer):
        if stochastic:
            mask_shape = ((1,) + inputs.shape[1:] if self.share_across_batch
                          else inputs.shape)
-            self._mask = (rng.uniform(size=mask_shape) < self.incl_prob)
+            self._mask = (self.rng.uniform(size=mask_shape) < self.incl_prob)
            return inputs * self._mask
        else:
            return inputs * self.incl_prob