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

white space fix.

上级 66344ce3
...@@ -281,7 +281,7 @@ def test_consistency_GPU_parallel(): ...@@ -281,7 +281,7 @@ def test_consistency_GPU_parallel():
samples = numpy.array(samples).flatten() samples = numpy.array(samples).flatten()
assert(numpy.allclose(samples, java_samples)) assert(numpy.allclose(samples, java_samples))
def basictest(f, steps, sample_size, prefix="", allow_01=False, inputs=[], def basictest(f, steps, sample_size, prefix="", allow_01=False, inputs=[],
target_avg=0.5, target_std=None, mean_rtol=0.01): target_avg=0.5, target_std=None, mean_rtol=0.01):
dt = 0.0 dt = 0.0
avg_std = 0.0 avg_std = 0.0
...@@ -344,7 +344,7 @@ def test_uniform(): ...@@ -344,7 +344,7 @@ def test_uniform():
R = MRG_RandomStreams(234, use_cuda=False) R = MRG_RandomStreams(234, use_cuda=False)
u = R.uniform(size=size) u = R.uniform(size=size)
f = theano.function(var_input, u, mode=mode) f = theano.function(var_input, u, mode=mode)
assert any([isinstance(node.op,theano.sandbox.rng_mrg.mrg_uniform) assert any([isinstance(node.op,theano.sandbox.rng_mrg.mrg_uniform)
for node in f.maker.env.toposort()]) for node in f.maker.env.toposort()])
theano.printing.debugprint(f) theano.printing.debugprint(f)
cpu_out = f(*input) cpu_out = f(*input)
...@@ -364,14 +364,14 @@ def test_uniform(): ...@@ -364,14 +364,14 @@ def test_uniform():
f = theano.function(var_input, theano.Out( f = theano.function(var_input, theano.Out(
theano.sandbox.cuda.basic_ops.gpu_from_host(u), theano.sandbox.cuda.basic_ops.gpu_from_host(u),
borrow=True), mode=mode_with_gpu) borrow=True), mode=mode_with_gpu)
assert any([isinstance(node.op,theano.sandbox.rng_mrg.GPU_mrg_uniform) assert any([isinstance(node.op,theano.sandbox.rng_mrg.GPU_mrg_uniform)
for node in f.maker.env.toposort()]) for node in f.maker.env.toposort()])
theano.printing.debugprint(f) theano.printing.debugprint(f)
gpu_out = numpy.asarray(f(*input)) gpu_out = numpy.asarray(f(*input))
print 'random?[:10]\n' print 'random?[:10]\n'
print gpu_out[0,0:10] print gpu_out[0,0:10]
print gpu_out[-1,0:10] print gpu_out[-1,-10:]
#print 'random?[-1,-10:]\n', gpu_out[-1,-10:] #print 'random?[-1,-10:]\n', gpu_out[-1,-10:]
basictest(f, steps, sample_size, prefix='mrg gpu', inputs=input) basictest(f, steps, sample_size, prefix='mrg gpu', inputs=input)
...@@ -403,7 +403,7 @@ def test_binomial(): ...@@ -403,7 +403,7 @@ def test_binomial():
sample_size = (500,50) sample_size = (500,50)
steps = int(1e3) steps = int(1e3)
rtol=0.01 rtol=0.01
x = tensor.matrix() x = tensor.matrix()
v = tensor.vector() v = tensor.vector()
for mean in [0.1, 0.5]: for mean in [0.1, 0.5]:
...@@ -507,7 +507,7 @@ def basic_multinomialtest(f, steps, sample_size, target_pvals, prefix="", mean_r ...@@ -507,7 +507,7 @@ def basic_multinomialtest(f, steps, sample_size, target_pvals, prefix="", mean_r
dt = 0.0 dt = 0.0
avg_pvals = numpy.zeros(target_pvals.shape, dtype=config.floatX) avg_pvals = numpy.zeros(target_pvals.shape, dtype=config.floatX)
for i in xrange(steps): for i in xrange(steps):
t0 = time.time() t0 = time.time()
ival = f() ival = f()
...@@ -516,7 +516,7 @@ def basic_multinomialtest(f, steps, sample_size, target_pvals, prefix="", mean_r ...@@ -516,7 +516,7 @@ def basic_multinomialtest(f, steps, sample_size, target_pvals, prefix="", mean_r
#ival = numpy.asarray(ival) #ival = numpy.asarray(ival)
avg_pvals += ival avg_pvals += ival
avg_pvals/= steps avg_pvals/= steps
print 'random?[:10]\n', f()[:10] print 'random?[:10]\n', f()[:10]
print prefix, 'mean', avg_pvals print prefix, 'mean', avg_pvals
print numpy.mean(abs(avg_pvals - target_pvals))# < mean_rtol, 'bad mean? %s %s' % (str(avg_pvals), str(target_pvals)) print numpy.mean(abs(avg_pvals - target_pvals))# < mean_rtol, 'bad mean? %s %s' % (str(avg_pvals), str(target_pvals))
...@@ -528,7 +528,7 @@ def test_multinomial(): ...@@ -528,7 +528,7 @@ def test_multinomial():
steps = 100 steps = 100
mode_ = mode mode_ = mode
if mode == 'FAST_COMPILE': if mode == 'FAST_COMPILE':
mode_ = 'FAST_RUN' mode_ = 'FAST_RUN'
if mode in ['DEBUG_MODE','DebugMode','FAST_COMPILE']: if mode in ['DEBUG_MODE','DebugMode','FAST_COMPILE']:
...@@ -545,7 +545,7 @@ def test_multinomial(): ...@@ -545,7 +545,7 @@ def test_multinomial():
m = R.multinomial(pvals=pvals, dtype=config.floatX) m = R.multinomial(pvals=pvals, dtype=config.floatX)
f = theano.function([], m, mode=mode_) f = theano.function([], m, mode=mode_)
theano.printing.debugprint(f) theano.printing.debugprint(f)
basic_multinomialtest(f, steps, sample_size, pvals, prefix='mrg ') basic_multinomialtest(f, steps, sample_size, pvals, prefix='mrg ')
sys.stdout.flush() sys.stdout.flush()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论