提交 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):
r *= s
if r > 6:
r = r/6 # chosen as fastest for rbm_benchmark
return r
return min(r, 30 * 256)
else:
if warn:
warnings.warn((
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论