提交 d20ad2a1 authored 作者: Frederic Bastien's avatar Frederic Bastien

Make test faster in FAST_COMPILE

上级 3d62cf6a
...@@ -60,11 +60,11 @@ def test_bn_feature_maps(): ...@@ -60,11 +60,11 @@ def test_bn_feature_maps():
return n * G + B return n * G + B
numpy.random.seed(1234) numpy.random.seed(1234)
X = 1 + numpy.random.random([10, 20, 4, 4]).astype('float32') X = 1 + numpy.random.random([2, 3, 4, 4]).astype('float32')
B = 1 + numpy.random.random([20]).astype('float32') B = 1 + numpy.random.random([3]).astype('float32')
G = 1 + numpy.random.random([20]).astype('float32') G = 1 + numpy.random.random([3]).astype('float32')
M = 1 + numpy.random.random([20]).astype('float32') M = 1 + numpy.random.random([3]).astype('float32')
V = 1 + numpy.random.random([20]).astype('float32') V = 1 + numpy.random.random([3]).astype('float32')
x = theano.tensor.tensor4('x') x = theano.tensor.tensor4('x')
b = theano.tensor.vector('b') b = theano.tensor.vector('b')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论