提交 060f121d authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Precise error message.

上级 5bf53d6d
...@@ -345,7 +345,7 @@ cudnn = cudnn_defs.get_definitions(version(raises=False)) ...@@ -345,7 +345,7 @@ cudnn = cudnn_defs.get_definitions(version(raises=False))
def get_precision(precision, inputs, for_grad=False): def get_precision(precision, inputs, for_grad=False):
common_dtype = theano.scalar.upcast(*[i.dtype for i in inputs]) common_dtype = theano.scalar.upcast(*[i.dtype for i in inputs])
if not common_dtype.startswith('float'): if not common_dtype.startswith('float'):
raise TypeError("cuDNN convolution only works on floats") raise TypeError("cuDNN convolution only works on real numbers")
if precision is None: if precision is None:
precision = theano.config.dnn.conv.precision precision = theano.config.dnn.conv.precision
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论