提交 2b33b542 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

GpuBatchedDot: tweak default stream_threshold

上级 14cf7b8d
...@@ -18,7 +18,7 @@ from theano.tensor import as_tensor_variable ...@@ -18,7 +18,7 @@ from theano.tensor import as_tensor_variable
class GpuBatchedDot(GpuOp): class GpuBatchedDot(GpuOp):
__props__ = ("stream_threshold",) __props__ = ("stream_threshold",)
def __init__(self, stream_threshold=128): def __init__(self, stream_threshold=650):
self.stream_threshold = stream_threshold self.stream_threshold = stream_threshold
def make_node(self, inp1, inp2): def make_node(self, inp1, inp2):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论