提交 6452bbd4 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Fix sandbox.cuda.dnn.dnn_conv3d border_mode check.

上级 726b4f4e
...@@ -1235,7 +1235,7 @@ def dnn_conv3d(img, kerns, border_mode='valid', subsample=(1, 1, 1), ...@@ -1235,7 +1235,7 @@ def dnn_conv3d(img, kerns, border_mode='valid', subsample=(1, 1, 1),
:warning: dnn_conv3d only works with cuDNN library 3.0 :warning: dnn_conv3d only works with cuDNN library 3.0
""" """
if border_mode == (0, 0): if border_mode == (0, 0, 0):
border_mode = 'valid' border_mode = 'valid'
# Establish dtype in which to perform the computation of the convolution # Establish dtype in which to perform the computation of the convolution
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论