提交 d962a575 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Fix test_rng_mrg test

上级 d8e47500
...@@ -90,7 +90,7 @@ def test_consistency_randomstreams(): ...@@ -90,7 +90,7 @@ def test_consistency_randomstreams():
for use_cuda in test_use_cuda: for use_cuda in test_use_cuda:
#print 'use_cuda =', use_cuda #print 'use_cuda =', use_cuda
samples = [] samples = []
rng = MRG_RandomStreams(seed=seed, use_cuda=False) rng = MRG_RandomStreams(seed=seed, use_cuda=use_cuda)
for i in range(n_streams): for i in range(n_streams):
stream_samples = [] stream_samples = []
u = rng.uniform(size=(n_substreams,), nstreams=n_substreams) u = rng.uniform(size=(n_substreams,), nstreams=n_substreams)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论