提交 e28ce73e authored 作者: Frederic's avatar Frederic

disable MRG_RandomStreams warning that it can't guess the number of stream.…

disable MRG_RandomStreams warning that it can't guess the number of stream. There is no explaination how to do this and confuse people.
上级 6ccc37b9
...@@ -1007,7 +1007,7 @@ class GPUA_mrg_uniform(GpuKernelBase, mrg_uniform_base): ...@@ -1007,7 +1007,7 @@ class GPUA_mrg_uniform(GpuKernelBase, mrg_uniform_base):
return (3, self.GpuKernelBase_version) return (3, self.GpuKernelBase_version)
def guess_n_streams(size, warn=True): def guess_n_streams(size, warn=False):
""" """
Return a guess at a good number of streams. Return a guess at a good number of streams.
...@@ -1128,7 +1128,7 @@ class MRG_RandomStreams(object): ...@@ -1128,7 +1128,7 @@ class MRG_RandomStreams(object):
return rval return rval
def n_streams(self, size): def n_streams(self, size):
return guess_n_streams(size, warn=True) return guess_n_streams(size)
def pretty_return(self, node_rstate, new_rstate, sample): def pretty_return(self, node_rstate, new_rstate, sample):
sample.rstate = node_rstate sample.rstate = node_rstate
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论