提交 6cd916b9 authored 作者: Faruk Ahmed's avatar Faruk Ahmed

fix floatX=float32 test bug

上级 68290a96
......@@ -89,6 +89,9 @@ def makeTester(name, op, gpu_op, cases, checks=None, mode_gpu=mode_with_gpu,
raise SkipTest(skip)
for testname, inputs in iteritems(cases):
for _ in xrange(len(inputs)):
if type(inputs[_]) is float:
inputs[_] = numpy.asarray(inputs[_], dtype=theano.config.floatX)
self.run_case(testname, inputs)
def run_case(self, testname, inputs):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论