Merge pull request #7 from mkudinov/feature-bugfix-implicit-casting

Bugfix: implicit casting int64 + float32 used to cause a type error. Cas...
pascanur · Oct 22, 2014 · 3691cbf · 3691cbf
2 parents da5ae37 + 457bf0f
commit 3691cbf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/groundhog/layers/rec_layers.py b/groundhog/layers/rec_layers.py
@@ -229,6 +229,7 @@ def step_fprop(self,
             W_hhs = self.W_hhs
             b_hhs = self.b_hhs
         preactiv = TT.dot(state_before, W_hhs[0]) +state_below
+        preactiv = TT.cast(preactiv,theano.config.floatX)
         h = self.activation[0](preactiv)
         if self.activ_noise and use_noise:
             h = h + self.trng.normal(h.shape, avg=0, std=self.activ_noise, dtype=h.dtype)
@@ -1065,6 +1066,7 @@ def step_fprop(self,
         else:
             state_before_ = state_before
         preactiv = TT.dot(state_before_, W_hh) + state_below
+        preactiv = TT.cast(preactiv,theano.config.floatX)
         h = self.activation(preactiv)
         if self.gating and gater_below:
             gater = self.gater_activation(TT.dot(state_before, G_hh) +