提交 93f95160 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test by casting to float32

上级 8975b933
...@@ -19,9 +19,12 @@ import theano.compile.mode ...@@ -19,9 +19,12 @@ import theano.compile.mode
mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu') mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu')
def my_rand(*shape):
return theano._asarray(numpy.random.rand(*shape),dtype='float32')
def test_dot(): def test_dot():
a = tcn.shared_constructor(numpy.random.rand(4,4), 'a') a = tcn.shared_constructor(my_rand(4,4), 'a')
b = tensor.fmatrix() b = tensor.fmatrix()
...@@ -31,7 +34,7 @@ def test_dot(): ...@@ -31,7 +34,7 @@ def test_dot():
print a0 print a0
for i, node in enumerate(f.maker.env.toposort()): for i, node in enumerate(f.maker.env.toposort()):
print i, node print i, node
bval = numpy.random.rand(4,4) bval = my_rand(4,4)
f(bval) f(bval)
print a.value print a.value
...@@ -39,7 +42,7 @@ def test_dot(): ...@@ -39,7 +42,7 @@ def test_dot():
def test_gemm(): def test_gemm():
a = tcn.shared_constructor(numpy.random.rand(4,4), 'a') a = tcn.shared_constructor(my_rand(4,4), 'a')
b = tensor.fmatrix('b') b = tensor.fmatrix('b')
c = tensor.fmatrix('c') c = tensor.fmatrix('c')
...@@ -50,8 +53,8 @@ def test_gemm(): ...@@ -50,8 +53,8 @@ def test_gemm():
print a0 print a0
for i, node in enumerate(f.maker.env.toposort()): for i, node in enumerate(f.maker.env.toposort()):
print i, node print i, node
bval = numpy.random.rand(4,4) bval = my_rand(4,4)
cval = numpy.random.rand(4,4) cval = my_rand(4,4)
f(bval,cval) f(bval,cval)
print a.value print a.value
...@@ -111,7 +114,7 @@ def test_downsample(): ...@@ -111,7 +114,7 @@ def test_downsample():
print 'test_downsample', shp, ds, ignore_border print 'test_downsample', shp, ds, ignore_border
ds_op = DownsampleFactorMax(ds, ignore_border=ignore_border) ds_op = DownsampleFactorMax(ds, ignore_border=ignore_border)
a = tcn.shared_constructor(numpy.random.rand(*shp), 'a') a = tcn.shared_constructor(my_rand(*shp), 'a')
f = pfunc([], ds_op(tensor.as_tensor_variable(a)), mode=mode_with_gpu) f = pfunc([], ds_op(tensor.as_tensor_variable(a)), mode=mode_with_gpu)
worked = False worked = False
for i, node in enumerate(f.maker.env.toposort()): for i, node in enumerate(f.maker.env.toposort()):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论