提交 52a7a4da authored 作者: ChienliMa's avatar ChienliMa

dtype should be same as config.floatX

上级 83d631f0
......@@ -317,11 +317,11 @@ class T_function(unittest.TestCase):
"""
A special testcase for logistic_sgd.py in Deep Learning Tutorial
"""
train_x = theano.shared(value=numpy.random.rand(10,10))
test_x = theano.shared(value=numpy.random.rand(10,10))
train_x = theano.shared(value=numpy.random.rand(10,10).astype(config.floatX))
test_x = theano.shared(value=numpy.random.rand(10,10).astype(config.floatX))
train_y = theano.shared(value=numpy.random.rand(10,1))
test_y = theano.shared(value=numpy.random.rand(10,1))
train_y = theano.shared(value=numpy.random.rand(10,1).astype(config.floatX))
test_y = theano.shared(value=numpy.random.rand(10,1).astype(config.floatX))
i = T.iscalar('index')
x = T.vector('x')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论