提交 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,6 +587,9 @@ class MRG_RandomStreams(object): ...@@ -587,6 +587,9 @@ class MRG_RandomStreams(object):
r = 1 r = 1
for s in size: for s in size:
r *= s r *= s
if r > 6:
return r/6 # chosen as fastest for rbm_benchmark
else:
return r return r
try: try:
rval = int(size) rval = int(size)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论