提交 a643adde authored 作者: Tanjay94's avatar Tanjay94

Modified warning message and fixed syntax error.

上级 022b93f8
......@@ -25,7 +25,7 @@ AddConfigVar('floatX',
)
AddConfigVar('warn_float64',
"Warning for float64 output",
"Warning, parts of the graph are in float64 which can't be run on the GPU",
BoolParam(False)
)
......
......@@ -575,7 +575,7 @@ class TensorVariable(_tensor_py_operators, Variable):
"""Subclass to add the tensor operators to the basic `Variable` class."""
def __init__(self, type):
if (config.warn_float64 and type = float64):
if (config.warn_float64 and type == float64):
print '''Warning, you are instanciating parts of your graph
with the float64 type (possibly by accident).
This means that those parts will not run on the GPU.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论