changed iround to round to keed the dtype to float. This use faster dot implementation and allow to move them to the gpu.

nouiz · nouiz · commit 8f908f3bfbf1 · 2010-04-30T14:50:34.000-04:00
diff --git a/code/rbm.py b/code/rbm.py
@@ -241,7 +241,7 @@ def get_pseudo_likelihood_cost(self, updates):
         bit_i_idx = theano.shared(value=0, name = 'bit_i_idx')
 
         # binarize the input image by rounding to nearest integer
-        xi = T.iround(self.input)
+        xi = T.round(self.input)
 
         # calculate free energy for the given bit configuration
         fe_xi = self.free_energy(xi)
