make the rbm code work in float32.

nouiz · nouiz · commit ee263d1aa5f0 · 2010-04-01T14:11:49.000-04:00
diff --git a/code/rbm.py b/code/rbm.py
@@ -84,7 +84,7 @@ def __init__(self, input=None, n_visible=784, n_hidden=500, \
         # initialize input layer for standalone RBM or layer0 of DBN
         self.input = input 
         if not input:
-            self.input = T.dmatrix('input')
+            self.input = T.matrix('input')
 
         self.W          = W
         self.hbias      = hbias
@@ -341,7 +341,7 @@ def test_rbm(learning_rate=0.1, training_epochs = 15,
     theano_rng = RandomStreams( rng.randint(2**30))
 
     # initialize storage for the persistent chain (state = hidden layer of chain)
-    persistent_chain = theano.shared(numpy.zeros((batch_size, 500)))
+    persistent_chain = theano.shared(numpy.zeros((batch_size, 500),dtype=theano.config.floatX))
 
     # construct the RBM class
     rbm = RBM( input = x, n_visible=28*28, \
diff --git a/code/test.py b/code/test.py
@@ -52,7 +52,7 @@ def speed():
 
     expected_times_64=numpy.asarray([  12.42313051,   28.09523582,  106.35365391,  153.62705898,  153.12310314,  425.09175086,  642.72824597,  652.52828193])
     expected_times_32=numpy.asarray([  13.29699826,   32.42813158,   68.03559947,  142.51003361,  107.00527334,
-  242.41721797,  490.40798998,float('NaN')])
+  242.41721797,  490.40798998, 528.88854146])
     def time_test(m,l,idx,f,**kwargs):
         if not to_exec[idx]:
             l[idx]=float('nan')
