提交 5e5def18 authored 作者: James Bergstra's avatar James Bergstra

mrg_rng - Changed heuristic for generating the number of threads. Changed it to

try and draw 6 samples per thread instead of one.
上级 1a78e93f
......@@ -587,7 +587,10 @@ class MRG_RandomStreams(object):
r = 1
for s in size:
r *= s
return r
if r > 6:
return r/6 # chosen as fastest for rbm_benchmark
else:
return r
try:
rval = int(size)
assert rval > 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论