提交 6341d126 authored 作者: James Bergstra's avatar James Bergstra

ENH: modify mrg to not allocate so many streams when computing large random tensors

This (a) wastes memory and (b) wastes a lot of time, because the code for initializing the random state is in Python and sometimes takes for freaking ever.
上级 572bf565
...@@ -618,7 +618,7 @@ def guess_n_streams(size, warn=True): ...@@ -618,7 +618,7 @@ def guess_n_streams(size, warn=True):
r *= s r *= s
if r > 6: if r > 6:
r = r/6 # chosen as fastest for rbm_benchmark r = r/6 # chosen as fastest for rbm_benchmark
return r return min(r, 30 * 256)
else: else:
if warn: if warn:
warnings.warn(( warnings.warn((
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论