提交 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
class GpuBatchedDot(GpuOp):
__props__ = ("stream_threshold",)
def __init__(self, stream_threshold=128):
def __init__(self, stream_threshold=650):
self.stream_threshold = stream_threshold
def make_node(self, inp1, inp2):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论